Skip to content

Commit d2f9f48

Browse files
authored
Skip keras local mode test on gpu and use random port for serving in the test (#134)
* Skip keras local mode test on gpu
1 parent 8e6c4f2 commit d2f9f48

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

test/integration/local/test_keras.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -25,21 +25,16 @@
2525
logging.basicConfig(level=logging.DEBUG)
2626

2727

28-
@pytest.fixture
29-
def local_mode_instance_type(processor):
30-
instance_type = 'local' if processor == 'cpu' else 'local_gpu'
31-
return instance_type
32-
33-
34-
def test_keras_training(sagemaker_local_session, docker_image, tmpdir, local_mode_instance_type):
28+
@pytest.mark.skip_gpu
29+
def test_keras_training(sagemaker_local_session, docker_image, tmpdir):
3530
entry_point = os.path.join(RESOURCE_PATH, 'keras_inception.py')
3631
output_path = 'file://{}'.format(tmpdir)
3732

3833
estimator = TensorFlow(
3934
entry_point=entry_point,
4035
role='SageMakerRole',
4136
train_instance_count=1,
42-
train_instance_type=local_mode_instance_type,
37+
train_instance_type='local',
4338
image_name=docker_image,
4439
sagemaker_session=sagemaker_local_session,
4540
model_dir='/opt/ml/model',

0 commit comments

Comments
 (0)