|
20 | 20 |
|
21 | 21 | from sagemaker import Session
|
22 | 22 | 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 | + |
23 | 28 |
|
24 | 29 | DEFAULT_REGION = 'us-west-2'
|
25 | 30 |
|
@@ -91,21 +96,21 @@ def chainer_version(request):
|
91 | 96 | return request.param
|
92 | 97 |
|
93 | 98 |
|
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]) |
95 | 100 | def tf_full_version(request):
|
96 | 101 | return request.param
|
97 | 102 |
|
98 | 103 |
|
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]) |
100 | 105 | def mxnet_full_version(request):
|
101 | 106 | return request.param
|
102 | 107 |
|
103 | 108 |
|
104 |
| -@pytest.fixture(scope='module', params=["0.4.0"]) |
| 109 | +@pytest.fixture(scope='module', params=[PYTORCH_VERSION]) |
105 | 110 | def pytorch_full_version(request):
|
106 | 111 | return request.param
|
107 | 112 |
|
108 | 113 |
|
109 |
| -@pytest.fixture(scope='module', params=['4.0.0', '4.1.0']) |
| 114 | +@pytest.fixture(scope='module', params=[CHAINER_VERSION]) |
110 | 115 | def chainer_full_version(request):
|
111 | 116 | return request.param
|
0 commit comments