Skip to content

Commit 4322915

Browse files
icywang86ruiEliza Zhang
authored andcommitted
Skip keras local mode test on gpu and use random port for serving in the test (aws#134)
* Skip keras local mode test on gpu
1 parent c776b4a commit 4322915

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/integration/local/test_keras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def test_keras_training(sagemaker_local_session, docker_image, tmpdir, framework
3535
entry_point=entry_point,
3636
role='SageMakerRole',
3737
train_instance_count=1,
38-
train_instance_type=local_mode_instance_type,
38+
train_instance_type='local',
3939
image_name=docker_image,
4040
sagemaker_session=sagemaker_local_session,
4141
model_dir='/opt/ml/model',

0 commit comments

Comments
 (0)