From b8956cdcb4cdf029d1efc5f57a4febf37bcd06ee Mon Sep 17 00:00:00 2001 From: Niket Kumar Bhumihar Date: Mon, 15 Apr 2024 20:33:00 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 625179587 --- checkpoint/orbax/checkpoint/utils.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/checkpoint/orbax/checkpoint/utils.py b/checkpoint/orbax/checkpoint/utils.py index 236f95db..96a866a1 100644 --- 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: