sagemaker 2.243.2__tar.gz → 2.243.4__tar.gz

{sagemaker-2.243.2 → sagemaker-2.243.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.243.2
+Version: 2.243.4
 Summary: Open source library for training and deploying models on Amazon SageMaker.
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk
 Author: Amazon Web Services
@@ -10,11 +10,11 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Natural Language :: English
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.8
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.9
 Requires-Dist: attrs<24,>=23.1.0
 Requires-Dist: boto3<2.0,>=1.35.75
 Requires-Dist: cloudpickle>=2.2.1
@@ -23,15 +23,15 @@ Requires-Dist: fastapi
 Requires-Dist: google-pasta
 Requires-Dist: importlib-metadata<7.0,>=1.4.0
 Requires-Dist: jsonschema
-Requires-Dist: numpy<2.0,>=1.9.0
+Requires-Dist: numpy==1.26.4
 Requires-Dist: omegaconf<=2.3,>=2.2
-Requires-Dist: packaging>=20.0
+Requires-Dist: packaging<25,>=23.0
 Requires-Dist: pandas
 Requires-Dist: pathos
 Requires-Dist: platformdirs
 Requires-Dist: protobuf<6.0,>=3.12
 Requires-Dist: psutil
-Requires-Dist: pyyaml~=6.0
+Requires-Dist: pyyaml>=6.0.1
 Requires-Dist: requests
 Requires-Dist: sagemaker-core<2.0.0,>=1.0.17
 Requires-Dist: schema
@@ -46,11 +46,11 @@ Requires-Dist: docker<8.0.0,>=5.0.2; extra == 'all'
 Requires-Dist: fastapi>=0.111.0; extra == 'all'
 Requires-Dist: nest-asyncio; extra == 'all'
 Requires-Dist: pyspark==3.3.2; extra == 'all'
-Requires-Dist: pyyaml<7,>=5.4.1; extra == 'all'
+Requires-Dist: pyyaml<7,>=6.0.1; extra == 'all'
 Requires-Dist: sagemaker-feature-store-pyspark-3-3; extra == 'all'
 Requires-Dist: sagemaker-mlflow>=0.1.0; extra == 'all'
 Requires-Dist: sagemaker-schema-inference-artifacts>=0.0.5; extra == 'all'
-Requires-Dist: scipy==1.10.1; extra == 'all'
+Requires-Dist: scipy==1.11.3; extra == 'all'
 Requires-Dist: urllib3<3.0.0,>=1.26.8; extra == 'all'
 Requires-Dist: uvicorn>=0.30.1; extra == 'all'
 Provides-Extra: feature-processor
@@ -64,16 +64,16 @@ Requires-Dist: sagemaker-schema-inference-artifacts>=0.0.5; extra == 'huggingfac
 Requires-Dist: uvicorn>=0.30.1; extra == 'huggingface'
 Provides-Extra: local
 Requires-Dist: docker<8.0.0,>=5.0.2; extra == 'local'
-Requires-Dist: pyyaml<7,>=5.4.1; extra == 'local'
+Requires-Dist: pyyaml<7,>=6.0.1; extra == 'local'
 Requires-Dist: urllib3<3.0.0,>=1.26.8; extra == 'local'
 Provides-Extra: sagemaker-mlflow
 Requires-Dist: sagemaker-mlflow>=0.1.0; extra == 'sagemaker-mlflow'
 Provides-Extra: scipy
-Requires-Dist: scipy==1.10.1; extra == 'scipy'
+Requires-Dist: scipy==1.11.3; extra == 'scipy'
 Provides-Extra: test
 Requires-Dist: accelerate<=0.27.0,>=0.24.1; extra == 'test'
 Requires-Dist: apache-airflow-providers-amazon==7.2.1; extra == 'test'
-Requires-Dist: apache-airflow==2.9.3; extra == 'test'
+Requires-Dist: apache-airflow==2.10.4; extra == 'test'
 Requires-Dist: attrs<24,>=23.1.0; extra == 'test'
 Requires-Dist: awslogs==0.14.0; extra == 'test'
 Requires-Dist: black==24.3.0; extra == 'test'
@@ -83,18 +83,19 @@ Requires-Dist: contextlib2==21.6.0; extra == 'test'
 Requires-Dist: coverage<6.2,>=5.2; extra == 'test'
 Requires-Dist: deepdiff>=8.0.0; extra == 'test'
 Requires-Dist: docker<8.0.0,>=5.0.2; extra == 'test'
-Requires-Dist: fabric==2.6.0; extra == 'test'
+Requires-Dist: fabric==3.2.2; extra == 'test'
 Requires-Dist: fastapi==0.115.4; extra == 'test'
 Requires-Dist: fastapi>=0.111.0; extra == 'test'
-Requires-Dist: flake8==4.0.1; extra == 'test'
+Requires-Dist: flake8==7.1.2; extra == 'test'
 Requires-Dist: huggingface-hub==0.26.2; extra == 'test'
 Requires-Dist: jinja2==3.1.6; extra == 'test'
 Requires-Dist: jsonpickle<4.0.0; extra == 'test'
+Requires-Dist: lexicon; extra == 'test'
 Requires-Dist: mlflow<2.13,>=2.12.2; extra == 'test'
 Requires-Dist: mock==4.0.3; extra == 'test'
 Requires-Dist: nbformat<6,>=5.9; extra == 'test'
 Requires-Dist: nest-asyncio; extra == 'test'
-Requires-Dist: numpy>=1.24.0; extra == 'test'
+Requires-Dist: numpy==1.26.4; extra == 'test'
 Requires-Dist: onnx==1.17.0; extra == 'test'
 Requires-Dist: opentelemetry-proto==1.27.0; extra == 'test'
 Requires-Dist: orderly-set<5.4.0; extra == 'test'
@@ -108,8 +109,8 @@ Requires-Dist: pytest-timeout==2.1.0; extra == 'test'
 Requires-Dist: pytest-xdist==2.4.0; extra == 'test'
 Requires-Dist: pytest==6.2.5; extra == 'test'
 Requires-Dist: pyvis==0.2.1; extra == 'test'
-Requires-Dist: pyyaml<7,>=5.4.1; extra == 'test'
-Requires-Dist: pyyaml==6.0; extra == 'test'
+Requires-Dist: pyyaml<7,>=6.0.1; extra == 'test'
+Requires-Dist: pyyaml>=6.0.1; extra == 'test'
 Requires-Dist: requests==2.32.2; extra == 'test'
 Requires-Dist: sagemaker-experiments==0.1.35; extra == 'test'
 Requires-Dist: sagemaker-feature-store-pyspark-3-3; extra == 'test'
@@ -117,10 +118,10 @@ Requires-Dist: sagemaker-mlflow>=0.1.0; extra == 'test'
 Requires-Dist: sagemaker-schema-inference-artifacts>=0.0.5; extra == 'test'
 Requires-Dist: schema==0.7.5; extra == 'test'
 Requires-Dist: scikit-learn==1.3.0; extra == 'test'
-Requires-Dist: scipy==1.10.1; extra == 'test'
+Requires-Dist: scipy==1.11.3; extra == 'test'
 Requires-Dist: stopit==1.1.2; extra == 'test'
-Requires-Dist: tensorboard<=2.15.2,>=2.9.0; extra == 'test'
-Requires-Dist: tensorflow<=2.15.1,>=2.9.0; extra == 'test'
+Requires-Dist: tensorboard<=2.18.0,>=2.16.2; extra == 'test'
+Requires-Dist: tensorflow<=2.18.0,>=2.16.2; extra == 'test'
 Requires-Dist: tox==3.24.5; extra == 'test'
 Requires-Dist: tritonclient[http]<2.37.0; extra == 'test'
 Requires-Dist: urllib3<3.0.0,>=1.26.8; extra == 'test'
@@ -224,10 +225,10 @@ Supported Python Versions
 
 SageMaker Python SDK is tested on:
 
-- Python 3.8
 - Python 3.9
 - Python 3.10
 - Python 3.11
+- Python 3.12
 
 Telemetry
 ~~~~~~~~~~~~~~~
@@ -321,9 +322,9 @@ Setup a Python environment, and install the dependencies listed in ``doc/require
 ::
 
     # conda
-    conda create -n sagemaker python=3.7
+    conda create -n sagemaker python=3.12
     conda activate sagemaker
-    conda install sphinx=3.1.1 sphinx_rtd_theme=0.5.0
+    conda install sphinx=5.1.1 sphinx_rtd_theme=0.5.0
 
     # pip
     pip install -r doc/requirements.txt

{sagemaker-2.243.2 → sagemaker-2.243.4}/README.rst

@@ -94,10 +94,10 @@ Supported Python Versions
 
 SageMaker Python SDK is tested on:
 
-- Python 3.8
 - Python 3.9
 - Python 3.10
 - Python 3.11
+- Python 3.12
 
 Telemetry
 ~~~~~~~~~~~~~~~
@@ -191,9 +191,9 @@ Setup a Python environment, and install the dependencies listed in ``doc/require
 ::
 
     # conda
-    conda create -n sagemaker python=3.7
+    conda create -n sagemaker python=3.12
     conda activate sagemaker
-    conda install sphinx=3.1.1 sphinx_rtd_theme=0.5.0
+    conda install sphinx=5.1.1 sphinx_rtd_theme=0.5.0
 
     # pip
     pip install -r doc/requirements.txt

sagemaker-2.243.4/VERSION

@@ -0,0 +1 @@
+2.243.4

{sagemaker-2.243.2 → sagemaker-2.243.4}/pyproject.toml

@@ -7,7 +7,7 @@ name = "sagemaker"
 dynamic = ["version", "optional-dependencies"]
 description = "Open source library for training and deploying models on Amazon SageMaker."
 readme = "README.rst"
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 authors = [
   { name = "Amazon Web Services" },
 ]
@@ -25,10 +25,10 @@ classifiers = [
   "License :: OSI Approved :: Apache Software License",
   "Natural Language :: English",
   "Programming Language :: Python",
-  "Programming Language :: Python :: 3.8",
   "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
   "attrs>=23.1.0,<24",
@@ -39,15 +39,15 @@ dependencies = [
   "google-pasta",
   "importlib-metadata>=1.4.0,<7.0",
   "jsonschema",
-  "numpy>=1.9.0,<2.0",
+  "numpy==1.26.4",
   "omegaconf>=2.2,<=2.3",
-  "packaging>=20.0",
+  "packaging>=23.0,<25",
   "pandas",
   "pathos",
   "platformdirs",
   "protobuf>=3.12,<6.0",
   "psutil",
-  "PyYAML~=6.0",
+  "PyYAML>=6.0.1",
   "requests",
   "sagemaker-core>=1.0.17,<2.0.0",
   "schema",

{sagemaker-2.243.2 → sagemaker-2.243.4}/requirements/extras/local_requirements.txt

@@ -1,3 +1,3 @@
 urllib3>=1.26.8,<3.0.0
 docker>=5.0.2,<8.0.0
-PyYAML>=5.4.1,<7
+PyYAML>=6.0.1,<7

sagemaker-2.243.4/requirements/extras/scipy_requirements.txt

@@ -0,0 +1 @@
+scipy==1.11.3

{sagemaker-2.243.2 → sagemaker-2.243.4}/requirements/extras/test_requirements.txt

@@ -1,7 +1,7 @@
 tox==3.24.5
-numpy>=1.24.0
+numpy==1.26.4
 build[virtualenv]==1.2.1
-flake8==4.0.1
+flake8==7.1.2
 pytest==6.2.5
 pytest-cov==3.0.0
 pytest-rerunfailures==10.2
@@ -14,10 +14,10 @@ awslogs==0.14.0
 black==24.3.0
 stopit==1.1.2
 # Update tox.ini to have correct version of airflow constraints file
-apache-airflow==2.9.3
+apache-airflow==2.10.4
 apache-airflow-providers-amazon==7.2.1
 attrs>=23.1.0,<24
-fabric==2.6.0
+fabric==3.2.2
 requests==2.32.2
 sagemaker-experiments==0.1.35
 Jinja2==3.1.6
@@ -26,13 +26,13 @@ pandas==1.4.4
 scikit-learn==1.3.0
 cloudpickle==2.2.1
 jsonpickle<4.0.0
-PyYAML==6.0
+PyYAML>=6.0.1
 # TODO find workaround
 xgboost>=1.6.2,<=1.7.6
 pillow>=10.0.1,<=11
 opentelemetry-proto==1.27.0
 protobuf==4.25.5
-tensorboard>=2.9.0,<=2.15.2
+tensorboard>=2.16.2,<=2.18.0
 transformers==4.48.0
 sentencepiece==0.1.99
 # https://github.com/triton-inference-server/server/issues/6246
@@ -42,7 +42,7 @@ onnx==1.17.0
 nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
 schema==0.7.5
-tensorflow>=2.9.0,<=2.15.1
+tensorflow>=2.16.2,<=2.18.0
 mlflow>=2.12.2,<2.13
 huggingface_hub==0.26.2
 uvicorn>=0.30.1
@@ -51,3 +51,4 @@ nest-asyncio
 sagemaker-mlflow>=0.1.0
 deepdiff>=8.0.0
 orderly-set<5.4.0
+lexicon

{sagemaker-2.243.2 → sagemaker-2.243.4}/src/sagemaker/config/config_schema.py

@@ -540,7 +540,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             "minItems": 0,
             "maxItems": 50,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html#sagemaker-CreateTrainingJob-request-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_CreateTrainingJob.html#sagemaker-CreateTrainingJob-request-Environment
         "environmentVariables": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,
@@ -553,13 +554,15 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             },
             "maxProperties": 48,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_S3DataSource.html#sagemaker-Type-S3DataSource-S3Uri
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_S3DataSource.html#sagemaker-Type-S3DataSource-S3Uri
         "s3Uri": {
             TYPE: "string",
             "pattern": "^(https|s3)://([^/]+)/?(.*)$",
             "maxLength": 1024,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html#sagemaker-Type-AlgorithmSpecification-ContainerEntrypoint
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_AlgorithmSpecification.html#sagemaker-Type-AlgorithmSpecification-ContainerEntrypoint
         "preExecutionCommand": {TYPE: "string", "pattern": r".*"},
         # Regex based on https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_PipelineDefinitionS3Location.html
         # except with an additional ^ and $ for the beginning and the end to closer align to
@@ -570,7 +573,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             "minLength": 3,
             "maxLength": 63,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MonitoringJobDefinition.html#sagemaker-Type-MonitoringJobDefinition-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_MonitoringJobDefinition.html#sagemaker-Type-MonitoringJobDefinition-Environment
         "environment-Length256-Properties50": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,
@@ -583,7 +587,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             },
             "maxProperties": 50,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTransformJob.html#sagemaker-CreateTransformJob-request-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_CreateTransformJob.html#sagemaker-CreateTransformJob-request-Environment
         "environment-Length10240-Properties16": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,
@@ -596,7 +601,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             },
             "maxProperties": 16,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ContainerDefinition.html#sagemaker-Type-ContainerDefinition-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_ContainerDefinition.html#sagemaker-Type-ContainerDefinition-Environment
         "environment-Length1024-Properties16": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,
@@ -609,7 +615,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             },
             "maxProperties": 16,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateProcessingJob.html#sagemaker-CreateProcessingJob-request-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        #   API_CreateProcessingJob.html#sagemaker-CreateProcessingJob-request-Environment
         "environment-Length256-Properties100": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,
@@ -622,7 +629,8 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
             },
             "maxProperties": 100,
         },
-        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html#sagemaker-CreateTrainingJob-request-Environment
+        # Regex is taken from https://docs.aws.amazon.com/sagemaker/latest/APIReference/
+        # API_CreateTrainingJob.html#sagemaker-CreateTrainingJob-request-Environment
         "environment-Length512-Properties48": {
             TYPE: OBJECT,
             ADDITIONAL_PROPERTIES: False,

{sagemaker-2.243.2 → sagemaker-2.243.4}/src/sagemaker/feature_store/dataset_builder.py

@@ -929,7 +929,7 @@ class DatasetBuilder:
                 selected_features += ", "
                 selected_features += ", ".join(
                     [
-                        f'fg_{i}."{feature_name}" as "{feature_name}.{(i+1)}"'
+                        f'fg_{i}."{feature_name}" as "{feature_name}.{(i + 1)}"'
                         for feature_name in feature_group.projected_feature_names
                     ]
                 )

{sagemaker-2.243.2 → sagemaker-2.243.4}/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json

@@ -14,6 +14,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -66,6 +67,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -118,6 +120,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -170,6 +173,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -222,6 +226,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -274,6 +279,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -326,6 +332,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -378,6 +385,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -430,6 +438,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -482,6 +491,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",
@@ -534,6 +544,7 @@
                 "registries": {
                     "af-south-1": "626614931356",
                     "ap-east-1": "871362719292",
+                    "ap-east-2": "975050140332",
                     "ap-northeast-1": "763104351884",
                     "ap-northeast-2": "763104351884",
                     "ap-northeast-3": "364406365360",