Skip to content

Commit 49b8510

Browse files
committed
fix flake8
1 parent 591ff66 commit 49b8510

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e
2828
patch.object(Session, "create_model", return_value="mock_model") as mock_create_model,
2929
patch.object(
3030
Session, "endpoint_from_production_variants"
31-
) as mock_endpoint_from_production_variants,
31+
) as mock_endpoint_from_production_variants
3232
):
3333
iam_client = sagemaker_session.boto_session.client("iam")
3434
role_arn = iam_client.get_role(RoleName=ROLE_NAME)["Role"]["Arn"]

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def test_create_lineage_when_no_lineage_exists_with_fg_only():
122122
FEATURE_GROUP_INPUT[1],
123123
FEATURE_GROUP_INPUT[0],
124124
],
125-
) as retrieve_feature_group_context_arns_method,
125+
) as retrieve_feature_group_context_arns_method
126126
patch.object(
127127
S3LineageEntityHandler,
128128
"retrieve_raw_data_artifact",

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def test_generate_requirements_exact_match(monkeypatch):
9999
patch("sagemaker.serve.detector.pickle_dependencies.subprocess.run") as subprocess_run,
100100
patch("sagemaker.serve.detector.pickle_dependencies.subprocess.Popen") as subprocess_popen,
101101
patch("builtins.open") as mocked_open,
102-
monkeypatch.context() as m,
102+
monkeypatch.context() as m
103103
):
104104
mock_run_stdout = MagicMock()
105105
mock_run_stdout.stdout = json.dumps(INSTALLED_PKG_JSON).encode("utf-8")

0 commit comments

Comments
 (0)