Skip to content

Commit e09693c

Browse files
authored
Fixed cpu isntance type for the estimator register test (#4811)
1 parent fa68f1f commit e09693c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_byo_estimator.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def test_byo_estimator(sagemaker_session, region, cpu_instance_type, training_se
109109

110110

111111
@pytest.mark.release
112-
def test_estimator_register_publish_training_details(sagemaker_session, region):
112+
def test_estimator_register_publish_training_details(sagemaker_session, region, cpu_instance_type):
113113

114114
bucket = sagemaker_session.default_bucket()
115115
prefix = "model-card-sample-notebook"
@@ -150,7 +150,7 @@ def test_estimator_register_publish_training_details(sagemaker_session, region):
150150
container,
151151
role="SageMakerRole",
152152
instance_count=1,
153-
instance_type="ml.m4.xlarge",
153+
instance_type=cpu_instance_type,
154154
output_path=output_location,
155155
sagemaker_session=sagemaker_session,
156156
)

0 commit comments

Comments
 (0)