Skip to content

Commit

Permalink
Add evaluation extra to google-cloud-aiplatform
Browse files Browse the repository at this point in the history
The scikit-learn is an optional dependency of the library from
google and apparently we are using the `evaluation` optional
feature of it. Adding the extra should make google provider
depends on scikit-learn.

Closes: #46258
  • Loading branch information
potiuk committed Jan 30, 2025
1 parent 3187141 commit 2ed07b4
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
2 changes: 1 addition & 1 deletion generated/provider_dependencies.json
Original file line number Diff line number Diff line change
Expand Up @@ -639,7 +639,7 @@
"google-api-python-client>=2.0.2",
"google-auth-httplib2>=0.0.1",
"google-auth>=2.29.0",
"google-cloud-aiplatform>=1.73.0,!=1.79.0",
"google-cloud-aiplatform[evaluation]>=1.73.0",
"google-cloud-alloydb>=0.4.0",
"google-cloud-automl>=2.12.0",
"google-cloud-batch>=0.13.0",
Expand Down
4 changes: 1 addition & 3 deletions providers/src/airflow/providers/google/provider.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -118,9 +118,7 @@ dependencies:
- google-api-python-client>=2.0.2
- google-auth>=2.29.0
- google-auth-httplib2>=0.0.1
# Google cloud aiplatform 1.79.0 has a dependency on sklearn which we don't use directly.
# See https://github.com/apache/airflow/issues/46258
- google-cloud-aiplatform>=1.73.0,!=1.79.0
- google-cloud-aiplatform[evaluation]>=1.73.0
- google-cloud-alloydb>=0.4.0
- google-cloud-automl>=2.12.0
# Excluded versions contain bug https://github.com/apache/airflow/issues/39541 which is resolved in 3.24.0
Expand Down

0 comments on commit 2ed07b4

Please sign in to comment.