diff --git a/tests/conftest.py b/tests/conftest.py index 7557c87fbe..34f5c5306d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -554,7 +554,9 @@ def _tf_py_version(tf_version, request): return "py38" if Version("2.8") <= version < Version("2.12"): return "py39" - return "py310" + if Version("2.12") <= version < Version("2.19"): + return "py310" + return "py312" @pytest.fixture(scope="module") @@ -597,7 +599,9 @@ def tf_full_py_version(tf_full_version): return "py38" if version < Version("2.12"): return "py39" - return "py310" + if version < Version("2.19"): + return "py310" + return "py312" @pytest.fixture(scope="module")