diff --git a/google/cloud/aiplatform/AutoMLTablesTrainingJob.py b/google/cloud/aiplatform/AutoMLTablesTrainingJob.py index 74f4495666..3a9a37cea7 100644 --- a/google/cloud/aiplatform/AutoMLTablesTrainingJob.py +++ b/google/cloud/aiplatform/AutoMLTablesTrainingJob.py @@ -220,4 +220,4 @@ def _model_upload_fail_string(self) -> str: return ( f"Training Pipeline {self.resource_name} is not configured to upload a " "Model." - ) \ No newline at end of file + ) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index 2d1484a86f..f586058c7a 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -808,7 +808,7 @@ def run( NotImplementedError more then one replica. ValueError if accelerator type is not valid. """ - + if self._has_run: raise RuntimeError("Custom Training has already run.") @@ -907,7 +907,7 @@ def run( managed_model = gca_model.Model( display_name=model_display_name, container_spec=container_spec ) - + return super()._run_job( training_task_definition=training_task_definition, training_task_inputs=training_task_inputs,