update for test

bencrabtree · bencrabtree · commit 988b2b5d8249 · 2025-04-03T19:32:47.000Z
diff --git a/src/sagemaker/jumpstart/utils.py b/src/sagemaker/jumpstart/utils.py
@@ -1653,6 +1653,7 @@ def remove_env_var_from_estimator_kwargs_if_model_access_config_present(
             del init_kwargs["environment"][
                 constants.SAGEMAKER_GATED_MODEL_S3_URI_TRAINING_ENV_VAR_KEY
             ]
+        if "accept_eula" in init_kwargs["environment"]:
             del init_kwargs["environment"]["accept_eula"]
 
 

Original file line number	Diff line number	Diff line change
`@@ -1653,6 +1653,7 @@ def remove_env_var_from_estimator_kwargs_if_model_access_config_present(`
`1653`	`1653`	`del init_kwargs["environment"][`
`1654`	`1654`	`constants.SAGEMAKER_GATED_MODEL_S3_URI_TRAINING_ENV_VAR_KEY`
`1655`	`1655`	`]`
	`1656`	`+ if "accept_eula" in init_kwargs["environment"]:`
`1656`	`1657`	`del init_kwargs["environment"]["accept_eula"]`
`1657`	`1658`
`1658`	`1659`