orbax icon indicating copy to clipboard operation
orbax copied to clipboard

Add additional support for TENSORSTORE_GCS_BACKEND environment variable

Open copybara-service[bot] opened this issue 3 months ago • 1 comments

Add additional support for TENSORSTORE_GCS_BACKEND environment variable

This allows the user to configure the Tensorstore GCS backend (i.e. with gcs for http or gcs_grpc for grpc) for additional places in the code that did not previously have support, including when using ocdbt.

This also refactors some previously duplicated code to parse the gcs bucket/path and generate the kvstore spec into the existing gcs_utils.py file.

copybara-service[bot] avatar Sep 22 '25 22:09 copybara-service[bot]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

google-cla[bot] avatar Sep 22 '25 22:09 google-cla[bot]