From 83f78dd2a1171b78dd67b2465ca3f7a0f2a3d09e Mon Sep 17 00:00:00 2001 From: vatsrahul1001 Date: Tue, 7 Apr 2026 22:49:48 +0530 Subject: [PATCH 1/2] Allow newer packages when testing stable release versions in compat tests --- .../install_airflow_and_providers.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/scripts/in_container/install_airflow_and_providers.py b/scripts/in_container/install_airflow_and_providers.py index 3118773721554..a1c665aa3a2fd 100755 --- a/scripts/in_container/install_airflow_and_providers.py +++ b/scripts/in_container/install_airflow_and_providers.py @@ -1127,9 +1127,13 @@ def _install_airflow_and_optionally_providers_together( "pip", "install", ] - if installation_spec.pre_release: + if installation_spec.pre_release or ( + installation_spec.airflow_distribution and "==" in installation_spec.airflow_distribution + ): console.print("[bright_blue]Allowing pre-release versions of airflow and providers") - base_install_cmd.extend(["--pre", "--exclude-newer", datetime.now().isoformat()]) + base_install_cmd.extend(["--exclude-newer", datetime.now().isoformat()]) + if installation_spec.pre_release: + base_install_cmd.append("--pre") if installation_spec.airflow_distribution: console.print( f"\n[bright_blue]Adding airflow distribution to installation: {installation_spec.airflow_distribution} " @@ -1224,9 +1228,13 @@ def _install_only_airflow_airflow_core_task_sdk_with_constraints( "pip", "install", ] - if installation_spec.pre_release: + if installation_spec.pre_release or ( + installation_spec.airflow_distribution and "==" in installation_spec.airflow_distribution + ): console.print("[bright_blue]Allowing pre-release versions of airflow and providers") - base_install_airflow_cmd.extend(["--pre", "--exclude-newer", datetime.now().isoformat()]) + base_install_airflow_cmd.extend(["--exclude-newer", datetime.now().isoformat()]) + if installation_spec.pre_release: + base_install_airflow_cmd.append("--pre") if installation_spec.airflow_distribution: console.print( f"\n[bright_blue]Installing airflow distribution: {installation_spec.airflow_distribution} with constraints" From 00aee9b3f25597fca84d8eb4d40071f577e62bfe Mon Sep 17 00:00:00 2001 From: vatsrahul1001 Date: Wed, 8 Apr 2026 00:01:04 +0530 Subject: [PATCH 2/2] address review comments --- scripts/in_container/install_airflow_and_providers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/in_container/install_airflow_and_providers.py b/scripts/in_container/install_airflow_and_providers.py index a1c665aa3a2fd..d008e24313911 100755 --- a/scripts/in_container/install_airflow_and_providers.py +++ b/scripts/in_container/install_airflow_and_providers.py @@ -1130,9 +1130,10 @@ def _install_airflow_and_optionally_providers_together( if installation_spec.pre_release or ( installation_spec.airflow_distribution and "==" in installation_spec.airflow_distribution ): - console.print("[bright_blue]Allowing pre-release versions of airflow and providers") + console.print("[bright_blue]Allowing install w/ cooldown as specific distribution requested.") base_install_cmd.extend(["--exclude-newer", datetime.now().isoformat()]) if installation_spec.pre_release: + console.print("[bright_blue]Allowing pre-release versions of airflow and providers") base_install_cmd.append("--pre") if installation_spec.airflow_distribution: console.print(