Ran linter

googleapis · May 3, 2021 · 8fb4947 · 8fb4947
1 parent 76116c7
commit 8fb4947
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 14 deletions.
diff --git a/samples/model-builder/create_and_import_dataset_video_sample_test.py b/samples/model-builder/create_and_import_dataset_video_sample_test.py
@@ -16,7 +16,6 @@
 from google.cloud.aiplatform import schema
 
 import create_and_import_dataset_video_sample
-
 import test_constants as constants
 
 

diff --git a/samples/model-builder/explain_tabular_sample.py b/samples/model-builder/explain_tabular_sample.py
@@ -14,10 +14,14 @@
 
 
 from google.cloud import aiplatform
+
 from typing import Dict
 
+
 #  [START aiplatform_sdk_explain_tabular_sample]
-def explain_tabular_sample(project: str, location: str, endpoint_id: str, instance_dict: Dict):
+def explain_tabular_sample(
+    project: str, location: str, endpoint_id: str, instance_dict: Dict
+):
 
     aiplatform.init(project=project, location=location)
 

diff --git a/samples/model-builder/explain_tabular_sample_test.py b/samples/model-builder/explain_tabular_sample_test.py
@@ -17,7 +17,9 @@
 import test_constants as constants
 
 
-def test_explain_tabular_sample(mock_sdk_init, mock_endpoint, mock_get_endpoint, mock_endpoint_explain):
+def test_explain_tabular_sample(
+    mock_sdk_init, mock_endpoint, mock_get_endpoint, mock_endpoint_explain
+):
 
     explain_tabular_sample.explain_tabular_sample(
         project=constants.PROJECT,
@@ -30,11 +32,8 @@ def test_explain_tabular_sample(mock_sdk_init, mock_endpoint, mock_get_endpoint,
         project=constants.PROJECT, location=constants.LOCATION
     )
 
-    mock_get_endpoint.assert_called_once_with(
-        constants.ENDPOINT_NAME,
-    )
+    mock_get_endpoint.assert_called_once_with(constants.ENDPOINT_NAME,)
 
     mock_endpoint_explain.assert_called_once_with(
-        instances=[constants.PREDICTION_TABULAR_INSTANCE],
-        parameters={}
-    )
+        instances=[constants.PREDICTION_TABULAR_INSTANCE], parameters={}
+    )
diff --git a/samples/model-builder/import_data_video_classification_sample_test.py b/samples/model-builder/import_data_video_classification_sample_test.py
@@ -17,7 +17,6 @@
 import pytest
 
 import import_data_video_classification_sample
-
 import test_constants as constants
 
 

diff --git a/samples/model-builder/upload_model_explain_tabular_managed_container_sample.py b/samples/model-builder/upload_model_explain_tabular_managed_container_sample.py
@@ -14,8 +14,9 @@
 
 
 from google.cloud import aiplatform
-from typing import Optional, Sequence, Tuple, Dict
-from google.cloud.aiplatform import explain
+
+from typing import Optional, Sequence, Dict
+
 
 #  [START aiplatform_sdk_upload_model_explain_tabular_managed_container_sample]
 def upload_model_explain_tabular_managed_container_sample(
@@ -34,8 +35,8 @@ def upload_model_explain_tabular_managed_container_sample(
     instance_schema_uri: Optional[str] = None,
     parameters_schema_uri: Optional[str] = None,
     prediction_schema_uri: Optional[str] = None,
-    explanation_metadata: Optional[explain.ExplanationMetadata] = None,
-    explanation_parameters: Optional[explain.ExplanationParameters] = None,
+    explanation_metadata: Optional[aiplatform.explain.ExplanationMetadata] = None,
+    explanation_parameters: Optional[aiplatform.explain.ExplanationParameters] = None,
     sync: bool = True,
 ):