use cloudpickle

GoogleCloudPlatform · Apr 11, 2024 · dcce68c · dcce68c
1 parent 55a3835
commit dcce68c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/people-and-planet-ai/image-classification/e2e_test.py b/people-and-planet-ai/image-classification/e2e_test.py
@@ -205,7 +205,7 @@ def test_train_model(
             f"--temp_location=gs://{bucket_name}/temp",
             "--requirements_file=requirements.txt",
             "--requirements_cache=skip",
-            "--experiments=use_sibling_sdk_workers",
+            "--pickle_library=cloudpickle",
         ],
         check=True,
     )