From 85e107f678ff2ce62ddfea920dc2ce545e915ad1 Mon Sep 17 00:00:00 2001 From: Yaning Liang Date: Tue, 26 Nov 2024 15:16:02 -0800 Subject: [PATCH] Internal PiperOrigin-RevId: 700474746 --- checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py | 1 + checkpoint/orbax/checkpoint/_src/path/snapshot/snapshot_test.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py b/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py index 8ce7c71ca..d9ac3e027 100644 --- a/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py +++ b/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py @@ -191,6 +191,7 @@ def save( multihost.process_index(), directory, ) + logging.info('test') if directory.exists(): if force: diff --git a/checkpoint/orbax/checkpoint/_src/path/snapshot/snapshot_test.py b/checkpoint/orbax/checkpoint/_src/path/snapshot/snapshot_test.py index 05cca0880..8bf4f9365 100644 --- a/checkpoint/orbax/checkpoint/_src/path/snapshot/snapshot_test.py +++ b/checkpoint/orbax/checkpoint/_src/path/snapshot/snapshot_test.py @@ -14,6 +14,7 @@ """test cases for colossus snapshot.""" +import logging import os from unittest import mock @@ -34,6 +35,7 @@ def __init__(self, *args, **kwargs): f.write('data') def test_create_snapshot(self): + logging.info('test') dst_dir = '/tmp/test/path/to/dest/' # dst_dir = epath.Path(self.create_tempdir(name='dest').full_path)