From e326590afaa2ffa25930bf403ee94dca88a5bab6 Mon Sep 17 00:00:00 2001 From: Claude Date: Wed, 16 Oct 2024 16:24:03 +0000 Subject: [PATCH] DO NOT MERGE: test cloudpickle default --- sdks/python/apache_beam/internal/pickler.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/internal/pickler.py b/sdks/python/apache_beam/internal/pickler.py index 79ebd16314bf..fc7cf15cd492 100644 --- a/sdks/python/apache_beam/internal/pickler.py +++ b/sdks/python/apache_beam/internal/pickler.py @@ -33,9 +33,10 @@ USE_CLOUDPICKLE = 'cloudpickle' USE_DILL = 'dill' -DEFAULT_PICKLE_LIB = USE_DILL +DEFAULT_PICKLE_LIB = USE_CLOUDPICKLE -desired_pickle_lib = dill_pickler +desired_pickle_lib = cloudpickle_pickler +dill_pickler.override_pickler_hooks(False) def dumps(o, enable_trace=True, use_zlib=False) -> bytes: