From 58c18ffeef9e002c3b0f384387705e86709b83c2 Mon Sep 17 00:00:00 2001
From: Jens Scheffler <95105677+jscheffl@users.noreply.github.com>
Date: Mon, 4 Nov 2024 23:03:59 +0100
Subject: [PATCH] Fix venv numpy example which needs to be 1.26 at least to be
 working in Python 3.12 (#43659)

---
 airflow/example_dags/example_branch_operator_decorator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/airflow/example_dags/example_branch_operator_decorator.py b/airflow/example_dags/example_branch_operator_decorator.py
index 59cb3b2919475..66cea26f391e2 100644
--- 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