Update airflow version

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",

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

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

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'