chore: add pandas to custom job test

PiperOrigin-RevId: 523428852
googleapis · Apr 11, 2023 · 8bf6477 · 8bf6477
1 parent cfb0bbd
commit 8bf6477
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/system/aiplatform/test_custom_job.py b/tests/system/aiplatform/test_custom_job.py
@@ -76,7 +76,7 @@ def test_from_local_script_prebuilt_container(self, shared_state):
             display_name=display_name,
             script_path=_LOCAL_TRAINING_SCRIPT_PATH,
             container_uri=_PREBUILT_CONTAINER_IMAGE,
-            requirements=["scikit-learn"],
+            requirements=["scikit-learn", "pandas"],
         )
         custom_job.run()
 
@@ -98,7 +98,7 @@ def test_from_local_script_custom_container(self, shared_state):
             display_name=display_name,
             script_path=_LOCAL_TRAINING_SCRIPT_PATH,
             container_uri=_CUSTOM_CONTAINER_IMAGE,
-            requirements=["scikit-learn"],
+            requirements=["scikit-learn", "pandas"],
         )
         custom_job.run()
 
@@ -127,7 +127,7 @@ def test_from_local_script_enable_autolog_prebuilt_container(self, shared_state)
             display_name=display_name,
             script_path=_LOCAL_TRAINING_SCRIPT_PATH,
             container_uri=_PREBUILT_CONTAINER_IMAGE,
-            requirements=["scikit-learn"],
+            requirements=["scikit-learn", "pandas"],
             enable_autolog=True,
         )
 
@@ -155,7 +155,7 @@ def test_from_local_script_enable_autolog_custom_container(self, shared_state):
             display_name=display_name,
             script_path=_LOCAL_TRAINING_SCRIPT_PATH,
             container_uri=_CUSTOM_CONTAINER_IMAGE,
-            requirements=["scikit-learn"],
+            requirements=["scikit-learn", "pandas"],
             enable_autolog=True,
         )