Fixed AutoMLVideo tests

googleapis · Aug 17, 2021 · 114560f · 114560f
1 parent bcc8ffe
commit 114560f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py
@@ -5672,15 +5672,21 @@ def _run(
         model_tbt.display_name = model_display_name or self._display_name
         model_tbt.labels = model_labels or self._labels
 
+        # AutoMLVideo does not support validation, so pass in '-' if any other filter split is provided.
+        validation_filter_split = (
+            "-"
+            if any([training_filter_split is not None, test_filter_split is not None])
+            else None
+        )
+
         return self._run_job(
             training_task_definition=training_task_definition,
             training_task_inputs=training_task_inputs_dict,
             dataset=dataset,
             training_fraction_split=training_fraction_split,
             test_fraction_split=test_fraction_split,
             training_filter_split=training_filter_split,
-            # AutoMLVideo does not support validation, so pass in '-'
-            validation_filter_split="-",
+            validation_filter_split=validation_filter_split,
             test_filter_split=test_filter_split,
             model=model_tbt,
         )