Ran linter

ivanmkc · ivanmkc · commit 8d474d61ff57 · 2021-04-30T14:31:08.000-04:00
diff --git a/samples/model-builder/create_and_import_dataset_image_sample_test.py b/samples/model-builder/create_and_import_dataset_image_sample_test.py
@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import create_and_import_dataset_image_sample
 
 from google.cloud.aiplatform import schema
 
-import create_and_import_dataset_image_sample
 import test_constants as constants
 
 
diff --git a/samples/model-builder/create_and_import_dataset_text_sample_test.py b/samples/model-builder/create_and_import_dataset_text_sample_test.py
@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import create_and_import_dataset_text_sample
 
 from google.cloud.aiplatform import schema
 
-import create_and_import_dataset_text_sample
 import test_constants as constants
 
 
diff --git a/samples/model-builder/deploy_model_with_automatic_resources_sample.py b/samples/model-builder/deploy_model_with_automatic_resources_sample.py
@@ -12,17 +12,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Dict, Optional, Sequence, Tuple
 
 from google.cloud import aiplatform
-from typing import Optional, Sequence, Tuple, Dict
-from google.cloud.aiplatform import explain
+
 
 #  [START aiplatform_sdk_deploy_model_with_automatic_resources_sample]
 def deploy_model_with_automatic_resources_sample(
     project,
     location,
     model_name: str,
-    endpoint: Optional["Endpoint"] = None,
+    endpoint: Optional[aiplatform.Endpoint] = None,
     deployed_model_display_name: Optional[str] = None,
     traffic_percentage: Optional[int] = 0,
     traffic_split: Optional[Dict[str, int]] = None,
diff --git a/samples/model-builder/deploy_model_with_dedicated_resources_sample.py b/samples/model-builder/deploy_model_with_dedicated_resources_sample.py
@@ -12,18 +12,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Dict, Optional, Sequence, Tuple
 
 from google.cloud import aiplatform
-from typing import Optional, Sequence, Tuple, Dict
 from google.cloud.aiplatform import explain
 
+
 #  [START aiplatform_sdk_deploy_model_with_dedicated_resources_sample]
 def deploy_model_with_dedicated_resources_sample(
     project,
     location,
     model_name: str,
     machine_type: str,
-    endpoint: Optional["Endpoint"] = None,
+    endpoint: Optional[aiplatform.Endpoint] = None,
     deployed_model_display_name: Optional[str] = None,
     traffic_percentage: Optional[int] = 0,
     traffic_split: Optional[Dict[str, int]] = None,
diff --git a/samples/model-builder/get_model_sample.py b/samples/model-builder/get_model_sample.py
@@ -14,15 +14,10 @@
 
 
 from google.cloud import aiplatform
-from typing import Optional, Sequence, Dict
-from google.cloud.aiplatform import explain
+
 
 #  [START aiplatform_sdk_get_model_sample]
-def get_model_sample(
-    project: str,
-    location: str,
-    model_name: str
-):
+def get_model_sample(project: str, location: str, model_name: str):
 
     aiplatform.init(project=project, location=location)
 
diff --git a/samples/model-builder/get_model_test.py b/samples/model-builder/get_model_test.py
@@ -29,6 +29,4 @@ def test_get_model_sample(mock_sdk_init, mock_init_model):
         project=constants.PROJECT, location=constants.LOCATION
     )
 
-    mock_init_model.assert_called_once_with(
-        model_name=constants.MODEL_NAME
-    )
+    mock_init_model.assert_called_once_with(model_name=constants.MODEL_NAME)
diff --git a/samples/model-builder/upload_model_sample.py b/samples/model-builder/upload_model_sample.py
@@ -12,11 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Dict, Optional, Sequence
 
 from google.cloud import aiplatform
-from typing import Optional, Sequence, Dict
 from google.cloud.aiplatform import explain
 
+
 #  [START aiplatform_sdk_upload_model_sample]
 def upload_model_sample(
     project: str,
diff --git a/samples/model-builder/upload_model_test.py b/samples/model-builder/upload_model_test.py
@@ -12,9 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
-import upload_model_sample
 import test_constants as constants
+import upload_model_sample
 
 
 def test_upload_model_sample(mock_sdk_init, mock_upload_model):
@@ -36,7 +35,7 @@ def test_upload_model_sample(mock_sdk_init, mock_upload_model):
         serving_container_environment_variables=constants.SERVING_CONTAINER_ENVIRONMENT_VARIABLES,
         serving_container_ports=constants.SERVING_CONTAINER_PORTS,
         explanation_metadata=constants.EXPLANATION_METADATA,
-        explanation_parameters=constants.EXPLANATION_PARAMETERS
+        explanation_parameters=constants.EXPLANATION_PARAMETERS,
     )
 
     mock_sdk_init.assert_called_once_with(

Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,4 @@ def test_get_model_sample(mock_sdk_init, mock_init_model):`
`29`	`29`	`project=constants.PROJECT, location=constants.LOCATION`
`30`	`30`	`)`
`31`	`31`
`32`		`- mock_init_model.assert_called_once_with(`
`33`		`- model_name=constants.MODEL_NAME`
`34`		`- )`
	`32`	`+ mock_init_model.assert_called_once_with(model_name=constants.MODEL_NAME)`