Fix --additional-extras parameter in Breeze command (apache#35710)

vchiapaikeo · Nov 18, 2023 · 0d278ca · 0d278ca
1 parent ed6fe24
commit 0d278ca
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/dev/breeze/src/airflow_breeze/params/build_ci_params.py b/dev/breeze/src/airflow_breeze/params/build_ci_params.py
@@ -94,7 +94,7 @@ def required_image_args(self) -> list[str]:
     @property
     def optional_image_args(self) -> list[str]:
         return [
-            "additional_airflow_extras",
+            "additional_extras",
             "additional_dev_apt_command",
             "additional_dev_apt_deps",
             "additional_dev_apt_env",
@@ -108,7 +108,7 @@ def optional_image_args(self) -> list[str]:
             "additional_dev_apt_command",
             "additional_dev_apt_deps",
             "additional_dev_apt_env",
-            "additional_airflow_extras",
+            "additional_extras",
             "additional_pip_install_flags",
             "additional_python_deps",
             "version_suffix_for_pypi",

diff --git a/dev/breeze/src/airflow_breeze/params/build_prod_params.py b/dev/breeze/src/airflow_breeze/params/build_prod_params.py
@@ -226,7 +226,7 @@ def required_image_args(self) -> list[str]:
     @property
     def optional_image_args(self) -> list[str]:
         return [
-            "additional_airflow_extras",
+            "additional_extras",
             "additional_dev_apt_command",
             "additional_dev_apt_deps",
             "additional_dev_apt_env",

diff --git a/dev/breeze/src/airflow_breeze/params/common_build_params.py b/dev/breeze/src/airflow_breeze/params/common_build_params.py
@@ -37,7 +37,7 @@ class CommonBuildParams:
     Common build parameters. Those parameters are common parameters for CI And PROD build.
     """
 
-    additional_airflow_extras: str = ""
+    additional_extras: str = ""
     additional_dev_apt_command: str = ""
     additional_dev_apt_deps: str = ""
     additional_dev_apt_env: str = ""