From 63a31ae5b3d7d950f8831e6db4b0ffc577cf63a6 Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 27 May 2025 11:59:09 -0700 Subject: [PATCH 1/4] Update Attrs version to widen support **Description** https://github.com/aws/sagemaker-python-sdk/issues/5075 **Testing Done** Running unit and integ tests Unit and integ tests passing indicate that this version upgrade does not break anything --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 17dfab3571..918e874b57 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ classifiers = [ "Programming Language :: Python :: 3.12", ] dependencies = [ - "attrs>=23.1.0,<24", + "attrs>=24,<26", "boto3>=1.35.75,<2.0", "cloudpickle>=2.2.1", "docker", From 836598767c442ae9d1b25f56275c8255e5f8ceba Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 27 May 2025 12:39:05 -0700 Subject: [PATCH 2/4] Update version in conda_in_process.yml --- src/sagemaker/serve/utils/conda_in_process.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/serve/utils/conda_in_process.yml b/src/sagemaker/serve/utils/conda_in_process.yml index 1f3fe322ef..d51754ec5a 100644 --- a/src/sagemaker/serve/utils/conda_in_process.yml +++ b/src/sagemaker/serve/utils/conda_in_process.yml @@ -8,7 +8,7 @@ dependencies: - fastapi>=0.111.0 - nest-asyncio - pip>=23.0.1 - - attrs>=23.1.0,<24 + - attrs>=24,<26 - boto3>=1.34.142,<2.0 - cloudpickle==2.2.1 - google-pasta From 73b85342ba8a59ca1b04bf8def68d0593b243bcb Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 27 May 2025 14:41:51 -0700 Subject: [PATCH 3/4] Update test requirements --- requirements/extras/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 8bdd7c8ae3..736217690b 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -17,7 +17,7 @@ stopit==1.1.2 apache-airflow==2.10.4 apache-airflow-providers-amazon==7.2.1 Flask-Limiter==3.11 -attrs>=23.1.0,<24 +attrs>=24,<26 fabric==3.2.2 requests==2.32.2 sagemaker-experiments==0.1.35 From 0e0ef3d03205f95b02608657909dc02094d1f8f6 Mon Sep 17 00:00:00 2001 From: Gokul A Date: Tue, 27 May 2025 20:09:49 -0700 Subject: [PATCH 4/4] MLFlow update version --- Tested by : Running unit and integ tests --- requirements/extras/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 736217690b..e4a9b3d09c 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -44,7 +44,7 @@ nbformat>=5.9,<6 accelerate>=0.24.1,<=0.27.0 schema==0.7.5 tensorflow>=2.16.2,<=2.18.0 -mlflow>=2.12.2,<2.13 +mlflow>=2.12.2,<3 huggingface_hub==0.26.2 uvicorn>=0.30.1 fastapi==0.115.4