Skip to content

Commit cd258f3

Browse files
muyangyuapplechanglan
authored andcommitted
change ENABLE_GCS_GRPC to op-in only
GitOrigin-RevId: 64bc782
1 parent 029cd03 commit cd258f3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

axlearn/common/array_serialization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,7 @@ async def _async_deserialize(
475475
# issues due to cross-cloud constraints. So we enable this optimization on Pathways
476476
# which only runs on GCP for now.
477477
context = serialization.TS_CONTEXT
478-
if running_on_pathways() or os.getenv("ENABLE_GCS_GRPC") == "true":
478+
if os.getenv("ENABLE_GCS_GRPC", "false") == "true":
479479
tensorstore_spec, context = use_gcs_grpc(tensorstore_spec)
480480

481481
t = await ts.open(tensorstore_spec, open=True, assume_metadata=False, context=context)

0 commit comments

Comments
 (0)