fix: tolerate vulnerable old model for integ test

evakravi · evakravi · commit d5dc7f804558 · 2024-02-26T21:57:53.000Z
diff --git a/tests/integ/sagemaker/jumpstart/estimator/test_jumpstart_estimator.py b/tests/integ/sagemaker/jumpstart/estimator/test_jumpstart_estimator.py
@@ -108,6 +108,7 @@ def test_gated_model_training_v1(setup):
         tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],
         environment={"accept_eula": "true"},
         max_run=259200,  # avoid exceeding resource limits
+        tolerate_vulnerable_model=True,
     )
 
     # uses ml.g5.12xlarge instance

Original file line number	Diff line number	Diff line change
`@@ -108,6 +108,7 @@ def test_gated_model_training_v1(setup):`
`108`	`108`	`tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],`
`109`	`109`	`environment={"accept_eula": "true"},`
`110`	`110`	`max_run=259200, # avoid exceeding resource limits`
	`111`	`+ tolerate_vulnerable_model=True,`
`111`	`112`	`)`
`112`	`113`
`113`	`114`	`# uses ml.g5.12xlarge instance`