DO NOT MERGE: test cloudpickle default

apache · Oct 16, 2024 · e326590 · e326590
1 parent 7311270
commit e326590
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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: