Fix keras & translate (#1339)

http://b/315458381 http://b/315753846
Kaggle · Dec 11, 2023 · a42ecff · a42ecff
1 parent 2ac180a
commit a42ecff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -403,7 +403,8 @@ RUN pip install annoy \
         google-cloud-bigquery \
         google-cloud-storage && \
     # Split these installations to avoid `pip._vendor.resolvelib.resolvers.ResolutionTooDeep: 200000`
-    pip install google-cloud-translate==3.* \
+    # TODO(b/315753846) Unpin translate package.
+    pip install google-cloud-translate==3.12.1 \
         google-cloud-language==2.* \
         google-cloud-videointelligence==2.* \
         google-cloud-vision==2.* \

diff --git a/tests/test_keras_core.py b/tests/test_keras_core.py
@@ -3,7 +3,7 @@
 import numpy as np
 import os
 
-os.environ["KERAS_BACKEND"] = "jax" 
+os.environ["KERAS_BACKEND"] = "tensorflow" 
 
 # Note that keras_core should only be imported after the backend
 # has been configured. The backend cannot be changed once the