Internal

PiperOrigin-RevId: 700474746
google · Dec 10, 2024 · e1af37a · e1af37a
1 parent c65b985
commit e1af37a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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)