Update docker build with TensorStore and list it as a dependency.

PiperOrigin-RevId: 436402469
tensorflow · Mar 22, 2022 · fe60d03 · fe60d03
1 parent e97a86a
commit fe60d03
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/lingvo/jax/README.md b/lingvo/jax/README.md
@@ -38,4 +38,6 @@ More specifically:
 
 *   [clu](http://pypi.org/project/clu/)
 
+*   [tensorstore](http://github.com/google/tensorstore)
+
 *   (optional) [MLPerf logging](http://github.com/mlcommons/logging)
diff --git a/lingvo/jax/pip_package/build.Dockerfile b/lingvo/jax/pip_package/build.Dockerfile
@@ -58,6 +58,7 @@ RUN for python in python3.7 python3.8 python3.9; do \
         scipy \
         sentencepiece \
         sympy \
+        tensorstore \
         twine && \
       $python -m pip install tensorflow tensorflow-datasets tensorflow-text; \
     done