Skip to content

Commit 6ecc49b

Browse files
sngyhantensorflower-gardener
authored andcommitted
Upgrade tensorflow version to 2.10.0
tensorflow_compression package which depend on tensorflow_probability requires tensorflow version >= 2.10 but the current version is 2.9.1. PiperOrigin-RevId: 474204911
1 parent 7790baa commit 6ecc49b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

ci/kokoro/gcp_ubuntu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
FROM ubuntu:20.04
3030

3131
ARG BAZEL_VERSION=4.2.2
32-
ARG TENSORFLOW_VERSION=2.9.1
32+
ARG TENSORFLOW_VERSION=2.10.0
3333

3434

3535
RUN apt-get update -y

ci/kokoro/run_bazel_unittests.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ set -o pipefail # Treat the failure of a command in a pipeline as error.
3434

3535
pip install --requirement "requirements.txt"
3636
# TODO(b/232345872): Not in list of requirements, but needed for EPR test.
37-
# The EPR test relies on a feature (PowerLawEntropyModel) introduced in 2.9.0.
38-
pip install tensorflow-compression~=2.9.1
37+
# The EPR test relies on a feature (PowerLawEntropyModel) introduced in 2.10.0.
38+
pip install tensorflow-compression~=2.10.0
3939

4040
# Run the tests.
4141
# Some tests requiring more RAM than the CI machine provides are disabled.

0 commit comments

Comments
 (0)