Skip to content

Commit af4402b

Browse files
authored
add py2 fixture for tf2.x test (#310)
1 parent cf213ca commit af4402b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/integration/sagemaker/test_mnist.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,7 @@ def test_tuning(sagemaker_session, ecr_image, instance_type, framework_version):
146146
tuner.wait()
147147

148148

149+
@pytest.mark.skip_py2_containers
149150
def test_smdebug(sagemaker_session, ecr_image, instance_type, framework_version):
150151
resource_path = os.path.join(os.path.dirname(__file__), '..', '..', 'resources')
151152
script = os.path.join(resource_path, 'mnist', 'mnist_smdebug.py')

0 commit comments

Comments
 (0)