Fixed AutoMLVideo tests

googleapis · Aug 17, 2021 · 705e976 · 705e976
1 parent bcc8ffe
commit 705e976
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py
@@ -5679,8 +5679,10 @@ def _run(
             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="-",
+            # AutoMLVideo does not support validation, so pass in '-' if any other split is provided.
+            validation_filter_split="-"
+            if any([training_filter_split is not None, test_filter_split is not None])
+            else None,
             test_filter_split=test_filter_split,
             model=model_tbt,
         )