Skip to content

Commit 26b4848

Browse files
cpgaffney1jax authors
authored andcommitted
Remove warning logs for primary_host/remote_storage "incompatibility".
PiperOrigin-RevId: 638022129
1 parent 8c55e8b commit 26b4848

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

jax/experimental/array_serialization/serialization.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -216,14 +216,6 @@ async def async_serialize(
216216
f'between processes. Serialization have failed for the array with '
217217
f'the path "{tensorstore_spec["kvstore"]["path"]}".')
218218

219-
if primary_host is None and is_remote_storage(tensorstore_spec):
220-
# Not strictly an error because users may manually split directories into
221-
# per-process subdirectories.
222-
logging.warning(
223-
'When primary_host is set to None and remote storage is used,'
224-
' serialization is not allowed, as this may lead to a race condition'
225-
' between processes.'
226-
)
227219
# 'metadata' may not be present at the top level (for example, if we are using
228220
# a 'cast' driver).
229221
if not _spec_has_metadata(tensorstore_spec):

0 commit comments

Comments
 (0)