amoghrajesh commented on code in PR #46242:
URL: https://github.com/apache/airflow/pull/46242#discussion_r1934085887
##########
generated/provider_dependencies.json:
##########
@@ -687,6 +687,7 @@
"proto-plus>=1.19.6",
"pyarrow>=14.0.1",
"python-slugify>=7.0.0",
+ "scikit-learn>=1.6.1",
Review Comment:
Seems to be a dependency in the vertexai code:
```
_ ERROR collecting
providers/tests/google/cloud/operators/vertex_ai/test_generative_model.py _
ImportError while importing test module
'/opt/airflow/providers/tests/google/cloud/operators/vertex_ai/test_generative_model.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/usr/local/lib/python3.9/importlib/__init__.py:127: in import_module
return _bootstrap._gcd_import(name[level:], package, level)
providers/tests/google/cloud/operators/vertex_ai/test_generative_model.py:32:
in <module>
from vertexai.preview.evaluation import MetricPromptTemplateExamples
/usr/local/lib/python3.9/site-packages/vertexai/preview/evaluation/__init__.py:20:
in <module>
from vertexai.preview.evaluation import (
/usr/local/lib/python3.9/site-packages/vertexai/preview/evaluation/autorater_utils.py:29:
in <module>
from sklearn import metrics
E ModuleNotFoundError: No module named 'sklearn'
```
--
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]
