Skip to content

Commit a0714ff

Browse files
committed
Fix codestyle
1 parent 9d806fa commit a0714ff

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,10 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e
8989
)
9090
# Verify the specific environment variables we care about
9191
actual_env = mock_create_model.call_args[1]["container_defs"]["Environment"]
92-
assert actual_env["OPTION_SPECULATIVE_DRAFT_MODEL"] == "/opt/ml/additional-model-data-sources/draft_model/"
92+
assert (
93+
actual_env["OPTION_SPECULATIVE_DRAFT_MODEL"]
94+
== "/opt/ml/additional-model-data-sources/draft_model/"
95+
)
9396
assert actual_env["SAGEMAKER_PROGRAM"] == "inference.py"
9497
assert actual_env["HF_MODEL_ID"] == "/opt/ml/model"
9598
mock_endpoint_from_production_variants.assert_called_once()

0 commit comments

Comments
 (0)