linting

Author: bencrabtree

src/sagemaker/session.py

@@ -3299,9 +3299,9 @@ def _map_training_config(
             training_job_definition["EnableNetworkIsolation"] = enable_network_isolation
 
         if encrypt_inter_container_traffic:
-            training_job_definition["EnableInterContainerTrafficEncryption"] = (
-                encrypt_inter_container_traffic
-            )
+            training_job_definition[
+                "EnableInterContainerTrafficEncryption"
+            ] = encrypt_inter_container_traffic
 
         if use_spot_instances:
             # use_spot_instances may be a Pipeline ParameterBoolean object

tests/unit/test_session.py

@@ -2482,9 +2482,9 @@ def boto_session_complete():
     boto_mock.client("logs").describe_log_streams.return_value = DEFAULT_LOG_STREAMS
     boto_mock.client("logs").get_log_events.side_effect = DEFAULT_LOG_EVENTS
     boto_mock.client("sagemaker").describe_training_job.return_value = COMPLETED_DESCRIBE_JOB_RESULT
-    boto_mock.client("sagemaker").describe_transform_job.return_value = (
-        COMPLETED_DESCRIBE_TRANSFORM_JOB_RESULT
-    )
+    boto_mock.client(
+        "sagemaker"
+    ).describe_transform_job.return_value = COMPLETED_DESCRIBE_TRANSFORM_JOB_RESULT
     return boto_mock
 
 
@@ -2504,9 +2504,9 @@ def boto_session_stopped():
     boto_mock.client("logs").describe_log_streams.return_value = DEFAULT_LOG_STREAMS
     boto_mock.client("logs").get_log_events.side_effect = DEFAULT_LOG_EVENTS
     boto_mock.client("sagemaker").describe_training_job.return_value = STOPPED_DESCRIBE_JOB_RESULT
-    boto_mock.client("sagemaker").describe_transform_job.return_value = (
-        STOPPED_DESCRIBE_TRANSFORM_JOB_RESULT
-    )
+    boto_mock.client(
+        "sagemaker"
+    ).describe_transform_job.return_value = STOPPED_DESCRIBE_TRANSFORM_JOB_RESULT
     return boto_mock