Ran linter

ivanmkc · ivanmkc · commit 12db685c2186 · 2021-05-03T17:01:11.000-04:00
diff --git a/samples/model-builder/create_training_pipeline_custom_container_job_sample.py b/samples/model-builder/create_training_pipeline_custom_container_job_sample.py
@@ -12,18 +12,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from google.cloud import aiplatform
 from typing import List, Optional, Union
 
+from google.cloud import aiplatform
+
 
 #  [START aiplatform_sdk_create_training_pipeline_custom_container_job_sample]
 def create_training_pipeline_custom_container_job_sample(
     project: str,
     location: str,
     staging_bucket: str,
     display_name: str,
-    container_uri: str,    
-    model_serving_container_image_uri: str,    
+    container_uri: str,
+    model_serving_container_image_uri: str,
     model_display_name: Optional[str] = None,
     args: Optional[List[Union[str, float, int]]] = None,
     replica_count: int = 1,
@@ -52,7 +53,7 @@ def create_training_pipeline_custom_container_job_sample(
         accelerator_count=accelerator_count,
         training_fraction_split=training_fraction_split,
         validation_fraction_split=validation_fraction_split,
-        test_fraction_split=test_fraction_split,        
+        test_fraction_split=test_fraction_split,
         sync=sync,
     )
 
diff --git a/samples/model-builder/create_training_pipeline_custom_container_job_test.py b/samples/model-builder/create_training_pipeline_custom_container_job_test.py
@@ -42,9 +42,11 @@ def test_create_training_pipeline_custom_container_job_sample(
     )
 
     mock_sdk_init.assert_called_once_with(
-        project=constants.PROJECT, location=constants.LOCATION, staging_bucket=constants.STAGING_BUCKET
+        project=constants.PROJECT,
+        location=constants.LOCATION,
+        staging_bucket=constants.STAGING_BUCKET,
     )
-    
+
     mock_init_custom_container_training_job.assert_called_once_with(
         display_name=constants.DISPLAY_NAME,
         container_uri=constants.CONTAINER_URI,
diff --git a/samples/model-builder/create_training_pipeline_custom_package_job_sample.py b/samples/model-builder/create_training_pipeline_custom_package_job_sample.py
@@ -12,9 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from google.cloud import aiplatform
 from typing import List, Optional, Union
 
+from google.cloud import aiplatform
+
 
 #  [START aiplatform_sdk_create_training_pipeline_custom_package_job_sample]
 def create_training_pipeline_custom_package_job_sample(
@@ -24,8 +25,8 @@ def create_training_pipeline_custom_package_job_sample(
     display_name: str,
     python_package_gcs_uri: str,
     python_module_name: str,
-    container_uri: str,    
-    model_serving_container_image_uri: str,    
+    container_uri: str,
+    model_serving_container_image_uri: str,
     model_display_name: Optional[str] = None,
     args: Optional[List[Union[str, float, int]]] = None,
     replica_count: int = 1,
@@ -56,7 +57,7 @@ def create_training_pipeline_custom_package_job_sample(
         accelerator_count=accelerator_count,
         training_fraction_split=training_fraction_split,
         validation_fraction_split=validation_fraction_split,
-        test_fraction_split=test_fraction_split,        
+        test_fraction_split=test_fraction_split,
         sync=sync,
     )
 
diff --git a/samples/model-builder/create_training_pipeline_custom_package_job_test.py b/samples/model-builder/create_training_pipeline_custom_package_job_test.py
@@ -29,7 +29,7 @@ def test_create_training_pipeline_custom_package_job_sample(
         staging_bucket=constants.STAGING_BUCKET,
         display_name=constants.DISPLAY_NAME,
         python_package_gcs_uri=constants.PYTHON_PACKAGE_GCS_URI,
-        python_module_name=constants.PYTHON_MODULE_NAME,        
+        python_module_name=constants.PYTHON_MODULE_NAME,
         container_uri=constants.CONTAINER_URI,
         args=constants.ARGS,
         model_serving_container_image_uri=constants.CONTAINER_URI,
@@ -44,9 +44,11 @@ def test_create_training_pipeline_custom_package_job_sample(
     )
 
     mock_sdk_init.assert_called_once_with(
-        project=constants.PROJECT, location=constants.LOCATION, staging_bucket=constants.STAGING_BUCKET
+        project=constants.PROJECT,
+        location=constants.LOCATION,
+        staging_bucket=constants.STAGING_BUCKET,
     )
-    
+
     mock_init_custom_package_training_job.assert_called_once_with(
         display_name=constants.DISPLAY_NAME,
         python_package_gcs_uri=constants.PYTHON_PACKAGE_GCS_URI,

Original file line number	Diff line number	Diff line change
`@@ -42,9 +42,11 @@ def test_create_training_pipeline_custom_container_job_sample(`
`42`	`42`	`)`
`43`	`43`
`44`	`44`	`mock_sdk_init.assert_called_once_with(`
`45`		`- project=constants.PROJECT, location=constants.LOCATION, staging_bucket=constants.STAGING_BUCKET`
	`45`	`+ project=constants.PROJECT,`
	`46`	`+ location=constants.LOCATION,`
	`47`	`+ staging_bucket=constants.STAGING_BUCKET,`
`46`	`48`	`)`
`47`		`-`
	`49`	`+`
`48`	`50`	`mock_init_custom_container_training_job.assert_called_once_with(`
`49`	`51`	`display_name=constants.DISPLAY_NAME,`
`50`	`52`	`container_uri=constants.CONTAINER_URI,`