From 38d0dd13cdf42f859eb61c66ab5eb62f30034f4a Mon Sep 17 00:00:00 2001
From: Ivy Zheng <ivyzhengx@gmail.com>
Date: Mon, 10 Oct 2022 18:03:44 -0700
Subject: [PATCH] Add Tensorstore back to required dependencies.

It was reversed back when user reported https://github.com/google/flax/issues/2341, but it was now fixed and verified.
---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index c0a508e8fe..fe7f0c0144 100644
--- 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",