Fix venv numpy example which needs to be 1.26 at least to be working …

…in Python 3.12 (#43659)
apache · Nov 4, 2024 · 58c18ff · 58c18ff
1 parent 07a080e
commit 58c18ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/example_dags/example_branch_operator_decorator.py b/airflow/example_dags/example_branch_operator_decorator.py
@@ -117,7 +117,7 @@ def some_ext_py_task():
         #       Run the example a second time and see that it re-uses it and is faster.
         VENV_CACHE_PATH = tempfile.gettempdir()
 
-        @task.branch_virtualenv(requirements=["numpy~=1.24.4"], venv_cache_path=VENV_CACHE_PATH)
+        @task.branch_virtualenv(requirements=["numpy~=1.26.0"], venv_cache_path=VENV_CACHE_PATH)
         def branching_virtualenv(choices) -> str:
             import random
 
@@ -137,7 +137,7 @@ def branching_virtualenv(choices) -> str:
         for option in options:
 
             @task.virtualenv(
-                task_id=f"venv_{option}", requirements=["numpy~=1.24.4"], venv_cache_path=VENV_CACHE_PATH
+                task_id=f"venv_{option}", requirements=["numpy~=1.26.0"], venv_cache_path=VENV_CACHE_PATH
             )
             def some_venv_task():
                 import numpy as np