gingeekrishna commented on issue #69323: URL: https://github.com/apache/airflow/issues/69323#issuecomment-4884866674
Opened PR #69399 with a fix. **Approach:** 1. : drop `[evaluation]` from the base dep (→ `google-cloud-aiplatform>=1.155.0`), add a new `vertex-ai-evaluation` optional extra that includes `google-cloud-aiplatform[evaluation]>=1.155.0` 2. `generative_model.py`: guard the `vertexai.preview.evaluation` import with `try/except ImportError`; `get_eval_task()` and `run_evaluation()` raise a clear `ImportError` pointing to `pip install apache-airflow-providers-google[vertex-ai-evaluation]` when the extra is missing Users of GCSHook/BigQueryHook etc. get no litellm/scikit-learn; users who need the Rapid Evaluation API add the explicit extra. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
