diff --git a/Dockerfile.ci b/Dockerfile.ci index bd5f476ca3aa0..f56de8e486635 100644 --- a/Dockerfile.ci +++ b/Dockerfile.ci @@ -782,6 +782,12 @@ if [[ ${SKIP_ENVIRONMENT_INITIALIZATION=} != "true" ]]; then installable_files+=( "${file}" ) fi done + if [[ ${USE_AIRFLOW_VERSION} != "wheel" && ${USE_AIRFLOW_VERSION} != "sdist" && ${USE_AIRFLOW_VERSION} != "none" ]]; then + echo + echo "${COLOR_BLUE}Also adding airflow in specified version ${USE_AIRFLOW_VERSION} to make sure it is not upgraded by >= limits${COLOR_RESET}" + echo + installable_files+=( "apache-airflow==${USE_AIRFLOW_VERSION}" ) + fi if (( ${#installable_files[@]} )); then pip install --root-user-action ignore "${installable_files[@]}" fi diff --git a/scripts/docker/entrypoint_ci.sh b/scripts/docker/entrypoint_ci.sh index a3918e54124d7..b33faaf74be4c 100755 --- a/scripts/docker/entrypoint_ci.sh +++ b/scripts/docker/entrypoint_ci.sh @@ -206,6 +206,12 @@ if [[ ${SKIP_ENVIRONMENT_INITIALIZATION=} != "true" ]]; then installable_files+=( "${file}" ) fi done + if [[ ${USE_AIRFLOW_VERSION} != "wheel" && ${USE_AIRFLOW_VERSION} != "sdist" && ${USE_AIRFLOW_VERSION} != "none" ]]; then + echo + echo "${COLOR_BLUE}Also adding airflow in specified version ${USE_AIRFLOW_VERSION} to make sure it is not upgraded by >= limits${COLOR_RESET}" + echo + installable_files+=( "apache-airflow==${USE_AIRFLOW_VERSION}" ) + fi if (( ${#installable_files[@]} )); then pip install --root-user-action ignore "${installable_files[@]}" fi