update tests with pyversion 311

Author: varunmoris

src/sagemaker/fw_utils.py

@@ -152,6 +152,7 @@
     "2.1.0",
     "2.1.2",
     "2.2.0",
+    "2.3.0",
     "2.3.1",
     "2.4.1",
 ]

src/sagemaker/serve/model_format/mlflow/constants.py

@@ -18,6 +18,7 @@
     "py38": "1.12.1",
     "py39": "1.13.1",
     "py310": "2.2.0",
+    "py311": "2.3.0",
 }
 MODEL_PACKAGE_ARN_REGEX = (
     r"^arn:aws:sagemaker:[a-z0-9\-]+:[0-9]{12}:model-package\/(.*?)(?:/(\d+))?$"

tests/conftest.py

@@ -356,7 +356,9 @@ def huggingface_training_compiler_pytorch_py_version(
 @pytest.fixture(scope="module")
 def huggingface_pytorch_latest_training_py_version(
     huggingface_training_pytorch_latest_version,
-):
+):    
+    if Version(huggingface_training_pytorch_latest_version) >= Version("2.3"):
+        return "py311"
     if Version(huggingface_training_pytorch_latest_version) >= Version("2.0"):
         return "py310"
     elif Version(huggingface_training_pytorch_latest_version) >= Version("1.13"):