diff --git a/pyproject.toml b/pyproject.toml index 918e874b57..1c4c11a8e6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,7 +37,7 @@ dependencies = [ "docker", "fastapi", "google-pasta", - "importlib-metadata>=1.4.0,<7.0", + "importlib-metadata>=1.4.0,<9.0", "jsonschema", "numpy==1.26.4", "omegaconf>=2.2,<3", diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 04d6c0522a..17481f99ff 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -14,9 +14,9 @@ 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.10.4 +apache-airflow==3.0.2 apache-airflow-providers-amazon==7.2.1 -Flask-Limiter==3.11 +Flask-Limiter==3.12 attrs>=24,<26 fabric==3.2.2 requests==2.32.2 diff --git a/src/sagemaker/serve/utils/conda_in_process.yml b/src/sagemaker/serve/utils/conda_in_process.yml index d51754ec5a..2346b405c4 100644 --- a/src/sagemaker/serve/utils/conda_in_process.yml +++ b/src/sagemaker/serve/utils/conda_in_process.yml @@ -15,7 +15,7 @@ dependencies: - numpy==1.26.4 - protobuf>=3.12,<5.0 - smdebug_rulesconfig==1.0.1 - - importlib-metadata>=1.4.0,<7.0 + - importlib-metadata>=1.4.0,<9.0 - packaging>=23.0,<25 - pandas - pathos diff --git a/tox.ini b/tox.ini index c47d206380..e4b9a61881 100644 --- a/tox.ini +++ b/tox.ini @@ -82,7 +82,7 @@ passenv = # Can be used to specify which tests to run, e.g.: tox -- -s commands = python -c "import os; os.system('install-custom-pkgs --install-boto-wheels')" - pip install 'apache-airflow==2.10.4' --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.10.4/constraints-3.9.txt" + pip install 'apache-airflow==3.0.2' --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-3.0.2/constraints-3.9.txt" pip install 'torch==2.3.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html' pip install 'torchvision==0.18.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html' pip install 'dill>=0.3.9'