From 444816e5e2a7b076a86b0da9b72fedc22aeba6fa Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 11 Mar 2025 15:51:42 -0700 Subject: [PATCH 1/2] Fixing Pytorch training python version in tests --- tests/conftest.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index 2c8dc2689f..76cc123ac7 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -254,6 +254,8 @@ def mxnet_eia_latest_py_version(): @pytest.fixture(scope="module", params=["py2", "py3"]) def pytorch_training_py_version(pytorch_training_version, request): + if Version(pytorch_training_version) >= Version("2.6"): + return "py312" if Version(pytorch_training_version) >= Version("2.3"): return "py311" elif Version(pytorch_training_version) >= Version("2.0"): From d612480a132accc149d73327c897af2fa5dcfc4f Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 11 Mar 2025 16:49:31 -0700 Subject: [PATCH 2/2] Updating Inference test handling --- tests/conftest.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 76cc123ac7..7557c87fbe 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -272,7 +272,9 @@ def pytorch_training_py_version(pytorch_training_version, request): @pytest.fixture(scope="module", params=["py2", "py3"]) def pytorch_inference_py_version(pytorch_inference_version, request): - if Version(pytorch_inference_version) >= Version("2.3"): + if Version(pytorch_inference_version) >= Version("2.6"): + return "py312" + elif Version(pytorch_inference_version) >= Version("2.3"): return "py311" elif Version(pytorch_inference_version) >= Version("2.0"): return "py310"