add pyyaml version constraint, remove py312 from docstring because there is no py312 image yet

Author: mollyheamazon

src/sagemaker/fw_utils.py

@@ -689,7 +689,7 @@ def validate_smdistributed(
         framework_name (str): A string representing the name of framework selected.
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
-        Ex: `py38, py39, py310, py311, py312`
+        Ex: `py38, py39, py310, py311`
         distribution (dict): A dictionary with information to enable distributed training.
             (Defaults to None if distributed training is not enabled.) For example:
 
@@ -762,7 +762,7 @@ def _validate_smdataparallel_args(
         framework_name (str): A string representing the name of framework selected. Ex: `tensorflow`
         framework_version (str): A string representing the framework version selected. Ex: `2.3.1`
         py_version (str): A string representing the python version selected.
-        Ex: `py38, py39, py310, py311, py312`
+        Ex: `py38, py39, py310, py311`
         distribution (dict): A dictionary with information to enable distributed training.
             (Defaults to None if distributed training is not enabled.) Ex:
 
@@ -846,7 +846,7 @@ def validate_distribution(
         framework_name (str): A string representing the name of framework selected.
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
-        Ex: `py38, py39, py310, py311, py312`
+        Ex: `py38, py39, py310, py311`
         image_uri (str): A string representing a Docker image URI.
         kwargs(dict): Additional kwargs passed to this function
 
@@ -1010,7 +1010,7 @@ def validate_torch_distributed_distribution(
                 }
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
-        Ex: `py38, py39, py310, py311, py312`
+        Ex: `py38, py39, py310, py311`
         image_uri (str): A string representing a Docker image URI.
         entry_point (str or PipelineVariable): The absolute or relative path to the local Python
             source file that should be executed as the entry point to
@@ -1162,7 +1162,7 @@ def validate_version_or_image_args(framework_version, py_version, image_uri):
     Args:
         framework_version (str): The version of the framework.
         py_version (str): A string representing the python version selected.
-        Ex: `py38, py39, py310, py311, py312`
+        Ex: `py38, py39, py310, py311`
         image_uri (str): The URI of the image.
 
     Raises:
@@ -1194,7 +1194,7 @@ def create_image_uri(
         instance_type (str): SageMaker instance type. Used to determine device
             type (cpu/gpu/family-specific optimized).
         framework_version (str): The version of the framework.
-        py_version (str): Optional. Python version Ex: `py38, py39, py310, py311, py312`.
+        py_version (str): Optional. Python version Ex: `py38, py39, py310, py311`.
             If not specified, image uri will not include a python component.
         account (str): AWS account that contains the image. (default:
             '520713654638')

src/sagemaker/processing.py

@@ -1465,7 +1465,7 @@ def __init__(
             instance_type (str or PipelineVariable): The type of EC2 instance to use for
                 processing, for example, 'ml.c4.xlarge'.
             py_version (str): Python version you want to use for executing your
-                model training code. Ex `py38, py39, py310, py311, py312`. Value
+                model training code. Ex `py38, py39, py310, py311`. Value
                 is ignored when ``image_uri`` is provided.
             image_uri (str or PipelineVariable): The URI of the Docker image to use for the
                 processing jobs (default: None).

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

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

tox.ini

@@ -86,7 +86,7 @@ commands =
     pip install 'torch==2.0.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'torchvision==0.15.2+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'dill>=0.3.8'
-    pip install --upgrade pip setuptools wheel
+    pip install "pyyaml>=6.0.1"
 
     pytest {posargs}
 deps = .[test]