jax

30a27c5a - Fix incorrect type annotation for tensorstore.Future

Fix incorrect type annotation for tensorstore.Future This fixes a type checking error that occurs when type stubs are added for tensorstore. PiperOrigin-RevId: 819276000