Move get_x_checkpoint functions to utils/checkpoint.py

Reviewed By: JKSenthil Differential Revision: D56450720
pytorch · Apr 26, 2024 · 888305f · 888305f
1 parent 4743897
commit 888305f
Show file tree

Hide file tree

Showing 7 changed files with 675 additions and 660 deletions.
diff --git a/tests/framework/callbacks/test_base_checkpointer.py b/tests/framework/callbacks/test_base_checkpointer.py
@@ -249,7 +249,7 @@ def test_restore_from_latest_empty_dir(self) -> None:
                 self.assertEqual(
                     log.output,
                     [
-                        f"WARNING:torchtnt.framework.callbacks._checkpoint_utils:Input dirpath doesn't contain any subdirectories: {temp_dir}"
+                        f"WARNING:torchtnt.utils.checkpoint:Input dirpath doesn't contain any subdirectories: {temp_dir}"
                     ],
                 )
                 self.assertFalse(restored)