add py312 to sagemaker tests (#4625)

Yadan-Wei · Yadan Wei · web-flow · commit 705618fba35a · 2025-03-10T14:37:14.000-07:00
Co-authored-by: Yadan Wei &lt;yadanwei@amazon.com&gt;
diff --git a/test/sagemaker_tests/pytorch/training/conftest.py b/test/sagemaker_tests/pytorch/training/conftest.py
@@ -133,7 +133,7 @@ def pytest_addoption(parser):
     parser.addoption("--framework-version", default="")
     parser.addoption(
         "--py-version",
-        choices=["2", "3", "37", "38", "39", "310", "311"],
+        choices=["2", "3", "37", "38", "39", "310", "311", "312"],
         default=str(sys.version_info.major),
     )
     parser.addoption("--processor", choices=["gpu", "cpu", "neuron", "neuronx"], default="cpu")

Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,7 @@ def pytest_addoption(parser):`
`133`	`133`	`parser.addoption("--framework-version", default="")`
`134`	`134`	`parser.addoption(`
`135`	`135`	`"--py-version",`
`136`		`- choices=["2", "3", "37", "38", "39", "310", "311"],`
	`136`	`+ choices=["2", "3", "37", "38", "39", "310", "311", "312"],`
`137`	`137`	`default=str(sys.version_info.major),`
`138`	`138`	`)`
`139`	`139`	`parser.addoption("--processor", choices=["gpu", "cpu", "neuron", "neuronx"], default="cpu")`