Skip to content

Commit 921f709

Browse files
author
Chuyang Deng
committed
move py_version fixture to test_tf_script_mode
1 parent f0e6b35 commit 921f709

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

tests/conftest.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -133,13 +133,6 @@ def custom_bucket_name(boto_session):
133133
return "{}-{}-{}".format(CUSTOM_BUCKET_NAME_PREFIX, region, account)
134134

135135

136-
@pytest.fixture(scope="module")
137-
def py_version():
138-
return (
139-
"py37" if tf_full_version == TensorFlow._LATEST_1X_VERSION else tests.integ.PYTHON_VERSION
140-
)
141-
142-
143136
@pytest.fixture(scope="module", params=["4.0", "4.0.0", "4.1", "4.1.0", "5.0", "5.0.0"])
144137
def chainer_version(request):
145138
return request.param

tests/integ/test_tf_script_mode.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,13 @@
3939
TAGS = [{"Key": "some-key", "Value": "some-value"}]
4040

4141

42+
@pytest.fixture(scope="module")
43+
def py_version(tf_full_version):
44+
return (
45+
"py37" if tf_full_version == TensorFlow._LATEST_1X_VERSION else tests.integ.PYTHON_VERSION
46+
)
47+
48+
4249
def test_mnist_with_checkpoint_config(
4350
sagemaker_session, instance_type, tf_full_version, py_version
4451
):

0 commit comments

Comments
 (0)