infra: black format unit tests (#2305)

Author: ajaykarpur

tests/unit/sagemaker/model/test_framework_model.py

@@ -56,7 +56,7 @@ def __init__(self, sagemaker_session, **kwargs):
             ROLE,
             ENTRY_POINT,
             sagemaker_session=sagemaker_session,
-            **kwargs
+            **kwargs,
         )
 
     def create_predictor(self, endpoint_name):
@@ -71,7 +71,7 @@ def __init__(self, sagemaker_session, entry_point, **kwargs):
             ROLE,
             entry_point=entry_point,
             sagemaker_session=sagemaker_session,
-            **kwargs
+            **kwargs,
         )
 
     def create_predictor(self, endpoint_name):

tests/unit/sagemaker/tensorflow/test_estimator.py

@@ -159,7 +159,7 @@ def _build_tf(
     py_version=None,
     instance_type=None,
     base_job_name=None,
-    **kwargs
+    **kwargs,
 ):
     return TensorFlow(
         entry_point=SCRIPT_PATH,
@@ -170,7 +170,7 @@ def _build_tf(
         instance_count=INSTANCE_COUNT,
         instance_type=instance_type if instance_type else INSTANCE_TYPE,
         base_job_name=base_job_name,
-        **kwargs
+        **kwargs,
     )
 
 

tests/unit/sagemaker/tensorflow/test_estimator_init.py

@@ -35,7 +35,7 @@ def _build_tf(sagemaker_session, **kwargs):
         role="dummy-role",
         instance_count=1,
         instance_type="ml.c4.xlarge",
-        **kwargs
+        **kwargs,
     )
 
 

tests/unit/test_amazon_estimator.py

@@ -86,7 +86,7 @@ def test_init_enable_network_isolation(sagemaker_session):
         num_components=55,
         sagemaker_session=sagemaker_session,
         enable_network_isolation=True,
-        **COMMON_ARGS
+        **COMMON_ARGS,
     )
     assert pca.num_components == 55
     assert pca.enable_network_isolation() is True
@@ -99,7 +99,7 @@ def test_init_all_pca_hyperparameters(sagemaker_session):
         subtract_mean=True,
         extra_components=33,
         sagemaker_session=sagemaker_session,
-        **COMMON_ARGS
+        **COMMON_ARGS,
     )
     assert pca.num_components == 55
     assert pca.algorithm_mode == "randomized"
@@ -112,7 +112,7 @@ def test_init_estimator_args(sagemaker_session):
         max_run=1234,
         sagemaker_session=sagemaker_session,
         data_location="s3://some-bucket/some-key/",
-        **COMMON_ARGS
+        **COMMON_ARGS,
     )
     assert pca.instance_type == COMMON_ARGS["instance_type"]
     assert pca.instance_count == COMMON_ARGS["instance_count"]
@@ -133,7 +133,7 @@ def test_data_location_does_not_call_default_bucket(sagemaker_session):
         num_components=2,
         sagemaker_session=sagemaker_session,
         data_location=data_location,
-        **COMMON_ARGS
+        **COMMON_ARGS,
     )
     assert pca.data_location == data_location
     assert not sagemaker_session.default_bucket.called
@@ -205,7 +205,7 @@ def test_fit_ndarray(time, sagemaker_session):
         num_components=55,
         sagemaker_session=sagemaker_session,
         data_location="s3://{}/key-prefix/".format(BUCKET_NAME),
-        **kwargs
+        **kwargs,
     )
     train = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 8.0], [44.0, 55.0, 66.0]]
     labels = [99, 85, 87, 2]
@@ -233,7 +233,7 @@ def test_fit_pass_experiment_config(sagemaker_session):
         num_components=55,
         sagemaker_session=sagemaker_session,
         data_location="s3://{}/key-prefix/".format(BUCKET_NAME),
-        **kwargs
+        **kwargs,
     )
     train = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 8.0], [44.0, 55.0, 66.0]]
     labels = [99, 85, 87, 2]

tests/unit/test_chainer.py

@@ -88,7 +88,7 @@ def _chainer_estimator(
     num_processes=None,
     process_slots_per_host=None,
     additional_mpi_options=None,
-    **kwargs
+    **kwargs,
 ):
     return Chainer(
         entry_point=SCRIPT_PATH,
@@ -103,7 +103,7 @@ def _chainer_estimator(
         num_processes=num_processes,
         process_slots_per_host=process_slots_per_host,
         additional_mpi_options=additional_mpi_options,
-        **kwargs
+        **kwargs,
     )
 
 

tests/unit/test_estimator.py

@@ -140,7 +140,7 @@ def create_model(
         vpc_config_override=vpc_utils.VPC_CONFIG_DEFAULT,
         enable_network_isolation=None,
         model_dir=None,
-        **kwargs
+        **kwargs,
     ):
         if enable_network_isolation is None:
             enable_network_isolation = self.enable_network_isolation()
@@ -151,7 +151,7 @@ def create_model(
             entry_point=entry_point,
             enable_network_isolation=enable_network_isolation,
             role=role,
-            **kwargs
+            **kwargs,
         )
 
     @classmethod
@@ -171,7 +171,7 @@ def __init__(self, sagemaker_session, entry_point=None, role=ROLE, **kwargs):
             role,
             entry_point or ENTRY_POINT,
             sagemaker_session=sagemaker_session,
-            **kwargs
+            **kwargs,
         )
 
     def create_predictor(self, endpoint_name):

tests/unit/test_fm.py

@@ -115,7 +115,7 @@ def test_all_hyperparameters(sagemaker_session):
         factors_init_scale=1.101,
         factors_init_sigma=1.202,
         factors_init_value=1.303,
-        **ALL_REQ_ARGS
+        **ALL_REQ_ARGS,
     )
     assert fm.hyperparameters() == dict(
         num_factors=str(ALL_REQ_ARGS["num_factors"]),

tests/unit/test_image.py

@@ -751,10 +751,10 @@ def test_ecr_login_needed(check_output):
     token_response = "AWS:%s" % token
     b64_token = base64.b64encode(token_response.encode("utf-8"))
     response = {
-        u"authorizationData": [
+        "authorizationData": [
             {
-                u"authorizationToken": b64_token,
-                u"proxyEndpoint": u"https://520713654638.dkr.ecr.us-east-1.amazonaws.com",
+                "authorizationToken": b64_token,
+                "proxyEndpoint": "https://520713654638.dkr.ecr.us-east-1.amazonaws.com",
             }
         ],
         "ResponseMetadata": {

tests/unit/test_ipinsights.py

@@ -103,7 +103,7 @@ def test_all_hyperparameters(sagemaker_session):
         random_negative_sampling_rate=5,
         shuffled_negative_sampling_rate=5,
         weight_decay=5.0,
-        **ALL_REQ_ARGS
+        **ALL_REQ_ARGS,
     )
     assert ipinsights.hyperparameters() == dict(
         num_entity_vectors=str(ALL_REQ_ARGS["num_entity_vectors"]),

tests/unit/test_job.py

@@ -107,7 +107,7 @@ def __init__(self, sagemaker_session, **kwargs):
             ROLE,
             SCRIPT_NAME,
             sagemaker_session=sagemaker_session,
-            **kwargs
+            **kwargs,
         )
 
     def prepare_container_def(self, instance_type, accelerator_type=None):