FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Merge pull request #637 from Kaggle/update-automl · RTDataScience/docker-python@b3ea7f2 · GitHub

Commit b3ea7f2

Browse files
authored
Merge pull request Kaggle#637 from Kaggle/update-automl
Update google-cloud-automl to latest version and fix broken test for Catalyst
2 parents b9699b0 + 6c38b61 commit b3ea7f2

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

‎Dockerfile‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -336,9 +336,7 @@ RUN pip install --upgrade cython && \
336336
pip install annoy==1.15.2 && \
337337
# Need to use CountEncoder from category_encoders before it's officially released
338338
pip install git+https://github.com/scikit-learn-contrib/categorical-encoding.git && \
339-
# TODO(b/143373325) Revert to 0.6.0 to avoid regression in 0.7.0. Go back to
340-
# latest when 0.8.0 is released.
341-
pip install google-cloud-automl==0.6.0 && \
339+
pip install google-cloud-automl && \
342340
# Newer version crashes (latest = 1.14.0) when running tensorflow.
343341
# python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py
344342
# which is loaded at startup.

‎tests/test_catalyst.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def test_mnist(self):
146146

147147
metrics = Safict.load("./logs/checkpoints/_metrics.json")
148148
metrics_flag1 = \
149-
metrics.get("train.2", "loss") < metrics.get("train.0", "loss")
149+
metrics.get("train.3", "loss") < metrics.get("train.1", "loss")
150150
metrics_flag2 = metrics.get("best", "loss") < 0.35
151151
self.assertTrue(metrics_flag1)
152152
self.assertTrue(metrics_flag2)

0 commit comments

Comments
 (0)

Back | FazBrowse Home | New Git URL