|
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