From 4b332ee5fed9f6e95ecc947a86f6fe37e1523139 Mon Sep 17 00:00:00 2001 From: Yu Date: Mon, 20 Aug 2018 15:14:57 -0700 Subject: [PATCH 1/2] Freeze framework integ test to latest version --- tests/conftest.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 7c379a9f06..3a9d9bffea 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -91,12 +91,12 @@ def chainer_version(request): return request.param -@pytest.fixture(scope='module', params=['1.4.1', '1.5.0', '1.6.0', '1.7.0', '1.8.0']) +@pytest.fixture(scope='module', params=['1.8.0']) def tf_full_version(request): return request.param -@pytest.fixture(scope='module', params=['0.12.1', '1.0.0', '1.1.0', '1.2.1']) +@pytest.fixture(scope='module', params=['1.2.1']) def mxnet_full_version(request): return request.param @@ -106,6 +106,6 @@ def pytorch_full_version(request): return request.param -@pytest.fixture(scope='module', params=['4.0.0', '4.1.0']) +@pytest.fixture(scope='module', params=['4.1.0']) def chainer_full_version(request): return request.param From dd5d200f98e186456e0693dabec94b7fdd446244 Mon Sep 17 00:00:00 2001 From: Yu Date: Mon, 20 Aug 2018 16:49:54 -0700 Subject: [PATCH 2/2] Use default version constant --- tests/conftest.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 3a9d9bffea..ddecdd687a 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -20,6 +20,11 @@ from sagemaker import Session from sagemaker.local import LocalSession +from sagemaker.chainer.defaults import CHAINER_VERSION +from sagemaker.pytorch.defaults import PYTORCH_VERSION +from sagemaker.mxnet.defaults import MXNET_VERSION +from sagemaker.tensorflow.defaults import TF_VERSION + DEFAULT_REGION = 'us-west-2' @@ -91,21 +96,21 @@ def chainer_version(request): return request.param -@pytest.fixture(scope='module', params=['1.8.0']) +@pytest.fixture(scope='module', params=[TF_VERSION]) def tf_full_version(request): return request.param -@pytest.fixture(scope='module', params=['1.2.1']) +@pytest.fixture(scope='module', params=[MXNET_VERSION]) def mxnet_full_version(request): return request.param -@pytest.fixture(scope='module', params=["0.4.0"]) +@pytest.fixture(scope='module', params=[PYTORCH_VERSION]) def pytorch_full_version(request): return request.param -@pytest.fixture(scope='module', params=['4.1.0']) +@pytest.fixture(scope='module', params=[CHAINER_VERSION]) def chainer_full_version(request): return request.param