fix flake8

Author: mollyheamazon

tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py

@@ -28,7 +28,7 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e
         patch.object(Session, "create_model", return_value="mock_model") as mock_create_model,
         patch.object(
             Session, "endpoint_from_production_variants"
-        ) as mock_endpoint_from_production_variants,
+        ) as mock_endpoint_from_production_variants
     ):
         iam_client = sagemaker_session.boto_session.client("iam")
         role_arn = iam_client.get_role(RoleName=ROLE_NAME)["Role"]["Arn"]

tests/unit/sagemaker/feature_store/feature_processor/lineage/test_feature_processor_lineage.py

@@ -122,7 +122,7 @@ def test_create_lineage_when_no_lineage_exists_with_fg_only():
                 FEATURE_GROUP_INPUT[1],
                 FEATURE_GROUP_INPUT[0],
             ],
-        ) as retrieve_feature_group_context_arns_method,
+        ) as retrieve_feature_group_context_arns_method
         patch.object(
             S3LineageEntityHandler,
             "retrieve_raw_data_artifact",

tests/unit/sagemaker/serve/detector/test_pickle_dependencies.py

@@ -99,7 +99,7 @@ def test_generate_requirements_exact_match(monkeypatch):
         patch("sagemaker.serve.detector.pickle_dependencies.subprocess.run") as subprocess_run,
         patch("sagemaker.serve.detector.pickle_dependencies.subprocess.Popen") as subprocess_popen,
         patch("builtins.open") as mocked_open,
-        monkeypatch.context() as m,
+        monkeypatch.context() as m
     ):
         mock_run_stdout = MagicMock()
         mock_run_stdout.stdout = json.dumps(INSTALLED_PKG_JSON).encode("utf-8")