Merge pull request #2520 from IvyZX:add-ts

PiperOrigin-RevId: 480389917
google · Oct 11, 2022 · ac56d59 · ac56d59
2 parents 5470cd3 + 38d0dd1
commit ac56d59
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -30,6 +30,7 @@
     "matplotlib",  # only needed for tensorboard export
     "msgpack",
     "optax",
+    "tensorstore",
     "rich>=11.1",
     "typing_extensions>=4.1.1",
     "PyYAML>=5.4.1",