From dcce68cdd4816f3f788fe858939635b40ad53476 Mon Sep 17 00:00:00 2001 From: David Cavazos Date: Thu, 11 Apr 2024 14:22:28 -0700 Subject: [PATCH] use cloudpickle --- people-and-planet-ai/image-classification/e2e_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/people-and-planet-ai/image-classification/e2e_test.py b/people-and-planet-ai/image-classification/e2e_test.py index b18be21f2d29..0cd146018036 100644 --- 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, )