Skip to content

Commit e96e629

Browse files
author
Deng
committed
change eia test instance type
1 parent 26f38e0 commit e96e629

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_pytorch_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def test_deploy_model_with_accelerator(sagemaker_session, cpu_instance_type):
134134
predictor = pytorch.deploy(
135135
initial_instance_count=1,
136136
instance_type=cpu_instance_type,
137-
accelerator_type="ml.eia1.large",
137+
accelerator_type="ml.eia2.medium",
138138
endpoint_name=endpoint_name,
139139
)
140140

0 commit comments

Comments
 (0)