Minor fixes

googleapis · Aug 30, 2021 · e169a4f · e169a4f
1 parent 8184673
commit e169a4f
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/google/cloud/aiplatform/datasets/column_names_dataset.py b/google/cloud/aiplatform/datasets/column_names_dataset.py
@@ -18,19 +18,12 @@
 
 import csv
 import logging
-
 from typing import List, Optional, Set
-
 from google.auth import credentials as auth_credentials
 
 from google.cloud import bigquery
 from google.cloud import storage
 
-from google.cloud.aiplatform import utils
-
-from typing import List, Optional
-
-from google.cloud.aiplatform import base
 from google.cloud.aiplatform import utils
 from google.cloud.aiplatform.datasets import _Dataset
 

diff --git a/tests/unit/aiplatform/test_automl_tabular_training_jobs.py b/tests/unit/aiplatform/test_automl_tabular_training_jobs.py
@@ -252,12 +252,6 @@ def mock_dataset_tabular():
     )
     ds.column_names = _TEST_TRAINING_COLUMN_NAMES
 
-    # TODO: Do this correctly
-    def mock_get_default_column_transformations(target_column: str,):
-        return (_TEST_TRAINING_COLUMN_TRANSFORMATIONS, _TEST_TRAINING_COLUMN_NAMES)
-
-    ds._get_default_column_transformations = mock_get_default_column_transformations
-
     yield ds