Skip to content

Commit 83bb841

Browse files
authored
Pin to version 1.0.1 for google-cloud-automl. (#876)
1 parent 959f7c2 commit 83bb841

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Dockerfile

+2-1
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,8 @@ RUN pip install --upgrade cython && \
291291
pip install annoy && \
292292
# Need to use CountEncoder from category_encoders before it's officially released
293293
pip install git+https://github.com/scikit-learn-contrib/categorical-encoding.git && \
294-
pip install google-cloud-automl && \
294+
# google-cloud-automl 2.0.0 introduced incompatible API changes, need to pin to 1.0.1
295+
pip install google-cloud-automl==1.0.1 && \
295296
# Newer version crashes (latest = 1.14.0) when running tensorflow.
296297
# python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py
297298
# which is loaded at startup.

0 commit comments

Comments
 (0)