From 114560f4db67375a21502197e8829122499c928d Mon Sep 17 00:00:00 2001 From: ivanmkc Date: Tue, 17 Aug 2021 14:42:00 -0400 Subject: [PATCH] Fixed AutoMLVideo tests --- google/cloud/aiplatform/training_jobs.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index 19794a4b8db..e84f531f3a5 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -5672,6 +5672,13 @@ 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, @@ -5679,8 +5686,7 @@ 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="-", + validation_filter_split=validation_filter_split, test_filter_split=test_filter_split, model=model_tbt, )