Fix keras & translate (#1339)

http://b/315458381
http://b/315753846

Author: Philmod

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.* \

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