Internal change.

PiperOrigin-RevId: 625179587
google · Apr 16, 2024 · b8956cd · b8956cd
1 parent a174452
commit b8956cd
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/checkpoint/orbax/checkpoint/utils.py b/checkpoint/orbax/checkpoint/utils.py
@@ -816,10 +816,7 @@ async def _async_is_locked(directory: epath.Path) -> bool:
   parent_dir_exists = await async_exists(directory)
   if not parent_dir_exists:
     raise ValueError(f'Parent directory {directory} does not exist.')
-  locked = await async_exists(lockdir(directory))
-  # TODO(b/333840560): Revert logging.
-  logging.info('step_name=%s, locked=%s', directory.name, locked)
-  return locked
+  return await async_exists(lockdir(directory))
 
 
 def is_locked(directory: epath.Path) -> bool: