chore: use ml.g5.2xlarge for integ test (#4741)

evakravi · web-flow · commit 908daa1577d8 · 2024-06-17T17:23:35.000-07:00
diff --git a/tests/integ/sagemaker/jumpstart/estimator/test_jumpstart_estimator.py b/tests/integ/sagemaker/jumpstart/estimator/test_jumpstart_estimator.py
@@ -174,6 +174,7 @@ def test_gated_model_training_v2(setup):
         tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],
         role=get_sm_session().get_caller_identity_arn(),
         sagemaker_session=get_sm_session(),
+        instance_type="ml.g5.2xlarge",
     )
 
     payload = {

Original file line number	Diff line number	Diff line change
`@@ -174,6 +174,7 @@ def test_gated_model_training_v2(setup):`
`174`	`174`	`tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],`
`175`	`175`	`role=get_sm_session().get_caller_identity_arn(),`
`176`	`176`	`sagemaker_session=get_sm_session(),`
	`177`	`+ instance_type="ml.g5.2xlarge",`
`177`	`178`	`)`
`178`	`179`
`179`	`180`	`payload = {`