@@ -350,7 +350,7 @@ def test_single_transformer_multiple_jobs(sagemaker_session, mxnet_full_version,
350
350
)
351
351
352
352
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 ):
354
354
data_path = os .path .join (DATA_DIR , "mxnet_mnist" )
355
355
script_path = os .path .join (data_path , "mnist.py" )
356
356
tags = [{"Key" : "some-tag" , "Value" : "value-for-tag" }]
@@ -359,7 +359,7 @@ def test_stop_transform_job(sagemaker_session, mxnet_full_version):
359
359
entry_point = script_path ,
360
360
role = "SageMakerRole" ,
361
361
train_instance_count = 1 ,
362
- train_instance_type = "ml.c4.xlarge" ,
362
+ train_instance_type = cpu_instance_type ,
363
363
sagemaker_session = sagemaker_session ,
364
364
framework_version = mxnet_full_version ,
365
365
)
@@ -381,7 +381,7 @@ def test_stop_transform_job(sagemaker_session, mxnet_full_version):
381
381
path = transform_input_path , key_prefix = transform_input_key_prefix
382
382
)
383
383
384
- transformer = mx .transformer (1 , "ml.m4.xlarge" , tags = tags )
384
+ transformer = mx .transformer (1 , cpu_instance_type , tags = tags )
385
385
transformer .transform (transform_input , content_type = "text/csv" )
386
386
387
387
time .sleep (15 )
0 commit comments