Skip to content

Commit a9717a0

Browse files
authored
Add TF CIFAR test to continuous testing set (#170)
Though this test has historically been relatively flaky, there was a recent change to the training steps that helped stabilize it, so it should be fine to include this test with our continuous testing. This test is important because it's currently our only one that tests distributed training and the only one to run on GPU instances.
1 parent de954b0 commit a9717a0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integ/test_tf_cifar.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import pickle
1515

1616
import numpy as np
17+
import pytest
1718

1819
from sagemaker.tensorflow import TensorFlow
1920
from tests.integ import DATA_DIR
@@ -30,6 +31,7 @@ def __call__(self, data):
3031
return pickle.dumps(data, protocol=2)
3132

3233

34+
@pytest.mark.continuous_testing
3335
def test_cifar(sagemaker_session, tf_full_version):
3436
with timeout(minutes=20):
3537
script_path = os.path.join(DATA_DIR, 'cifar_10', 'source')

0 commit comments

Comments
 (0)