From a42ecffd7b98e0209d94ae74eb9de9bcb891cdb4 Mon Sep 17 00:00:00 2001 From: Philippe Modard Date: Mon, 11 Dec 2023 15:04:58 +0100 Subject: [PATCH] Fix keras & translate (#1339) http://b/315458381 http://b/315753846 --- Dockerfile.tmpl | 3 ++- tests/test_keras_core.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 848ae40c..0c7365cf 100644 --- 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 index bedeefb0..29c8cd90 100644 --- 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