Skip to content

Commit 3739e5b

Browse files
authored
change: use cpu_instance_type fixture for stop_transform_job test (#1022)
1 parent 379ceac commit 3739e5b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/integ/test_transformer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -350,7 +350,7 @@ def test_single_transformer_multiple_jobs(sagemaker_session, mxnet_full_version,
350350
)
351351

352352

353-
def test_stop_transform_job(sagemaker_session, mxnet_full_version):
353+
def test_stop_transform_job(sagemaker_session, mxnet_full_version, cpu_instance_type):
354354
data_path = os.path.join(DATA_DIR, "mxnet_mnist")
355355
script_path = os.path.join(data_path, "mnist.py")
356356
tags = [{"Key": "some-tag", "Value": "value-for-tag"}]
@@ -359,7 +359,7 @@ def test_stop_transform_job(sagemaker_session, mxnet_full_version):
359359
entry_point=script_path,
360360
role="SageMakerRole",
361361
train_instance_count=1,
362-
train_instance_type="ml.c4.xlarge",
362+
train_instance_type=cpu_instance_type,
363363
sagemaker_session=sagemaker_session,
364364
framework_version=mxnet_full_version,
365365
)
@@ -381,7 +381,7 @@ def test_stop_transform_job(sagemaker_session, mxnet_full_version):
381381
path=transform_input_path, key_prefix=transform_input_key_prefix
382382
)
383383

384-
transformer = mx.transformer(1, "ml.m4.xlarge", tags=tags)
384+
transformer = mx.transformer(1, cpu_instance_type, tags=tags)
385385
transformer.transform(transform_input, content_type="text/csv")
386386

387387
time.sleep(15)

0 commit comments

Comments
 (0)