Moved automl tables training job to training_jobs.py

googleapis · Nov 17, 2020 · e709025 · e709025
1 parent 2802dc8
commit e709025
Show file tree

Hide file tree

Showing 4 changed files with 215 additions and 227 deletions.
diff --git a/google/cloud/aiplatform/AutoMLTablesTrainingJob.py b/google/cloud/aiplatform/AutoMLTablesTrainingJob.py
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -21,8 +21,10 @@
 from google.cloud.aiplatform.datasets import Dataset
 from google.cloud.aiplatform.models import Endpoint
 from google.cloud.aiplatform.models import Model
-from google.cloud.aiplatform.training_jobs import CustomTrainingJob
-from google.cloud.aiplatform.AutoMLTablesTrainingJob import AutoMLTablesTrainingJob
+from google.cloud.aiplatform.training_jobs import (
+    CustomTrainingJob,
+    AutoMLTablesTrainingJob,
+)
 
 """
 Usage: