From 5b371330ef932d65bac61987284a4836fd3c684c Mon Sep 17 00:00:00 2001 From: Jim Plotts Date: Fri, 18 Sep 2020 11:55:51 -0400 Subject: [PATCH 1/2] Pin to version 1.0.1 for google-cloud-automl. --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 43a04ccd..bb3f04b1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -291,7 +291,7 @@ RUN pip install --upgrade cython && \ pip install annoy && \ # Need to use CountEncoder from category_encoders before it's officially released pip install git+https://github.com/scikit-learn-contrib/categorical-encoding.git && \ - pip install google-cloud-automl && \ + pip install google-cloud-automl==1.0.1 && \ # Newer version crashes (latest = 1.14.0) when running tensorflow. # python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py # which is loaded at startup. From e26497e7fa2e8dd805b15c77d9141df55e3269f9 Mon Sep 17 00:00:00 2001 From: Jim Plotts Date: Fri, 18 Sep 2020 12:07:09 -0400 Subject: [PATCH 2/2] Add comment explaining pin. --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index bb3f04b1..2d9c5d5e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -291,6 +291,7 @@ RUN pip install --upgrade cython && \ pip install annoy && \ # Need to use CountEncoder from category_encoders before it's officially released pip install git+https://github.com/scikit-learn-contrib/categorical-encoding.git && \ + # google-cloud-automl 2.0.0 introduced incompatible API changes, need to pin to 1.0.1 pip install google-cloud-automl==1.0.1 && \ # Newer version crashes (latest = 1.14.0) when running tensorflow. # python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py