We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9d806fa commit a0714ffCopy full SHA for a0714ff
tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py
@@ -89,7 +89,10 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e
89
)
90
# Verify the specific environment variables we care about
91
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/"
+ assert (
93
+ actual_env["OPTION_SPECULATIVE_DRAFT_MODEL"]
94
+ == "/opt/ml/additional-model-data-sources/draft_model/"
95
+ )
96
assert actual_env["SAGEMAKER_PROGRAM"] == "inference.py"
97
assert actual_env["HF_MODEL_ID"] == "/opt/ml/model"
98
mock_endpoint_from_production_variants.assert_called_once()
0 commit comments