Skip to content

Commit d7665de

Browse files
authored
Freeze framework integ test to latest version (#359)
1 parent 80fa52e commit d7665de

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

tests/conftest.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,11 @@
2020

2121
from sagemaker import Session
2222
from sagemaker.local import LocalSession
23+
from sagemaker.chainer.defaults import CHAINER_VERSION
24+
from sagemaker.pytorch.defaults import PYTORCH_VERSION
25+
from sagemaker.mxnet.defaults import MXNET_VERSION
26+
from sagemaker.tensorflow.defaults import TF_VERSION
27+
2328

2429
DEFAULT_REGION = 'us-west-2'
2530

@@ -91,21 +96,21 @@ def chainer_version(request):
9196
return request.param
9297

9398

94-
@pytest.fixture(scope='module', params=['1.4.1', '1.5.0', '1.6.0', '1.7.0', '1.8.0'])
99+
@pytest.fixture(scope='module', params=[TF_VERSION])
95100
def tf_full_version(request):
96101
return request.param
97102

98103

99-
@pytest.fixture(scope='module', params=['0.12.1', '1.0.0', '1.1.0', '1.2.1'])
104+
@pytest.fixture(scope='module', params=[MXNET_VERSION])
100105
def mxnet_full_version(request):
101106
return request.param
102107

103108

104-
@pytest.fixture(scope='module', params=["0.4.0"])
109+
@pytest.fixture(scope='module', params=[PYTORCH_VERSION])
105110
def pytorch_full_version(request):
106111
return request.param
107112

108113

109-
@pytest.fixture(scope='module', params=['4.0.0', '4.1.0'])
114+
@pytest.fixture(scope='module', params=[CHAINER_VERSION])
110115
def chainer_full_version(request):
111116
return request.param

0 commit comments

Comments
 (0)