Skip to content

Commit 888305f

Browse files
diego-urgellfacebook-github-bot
authored andcommitted
Move get_x_checkpoint functions to utils/checkpoint.py
Reviewed By: JKSenthil Differential Revision: D56450720
1 parent 4743897 commit 888305f

File tree

7 files changed

+675
-660
lines changed

7 files changed

+675
-660
lines changed

tests/framework/callbacks/test_base_checkpointer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def test_restore_from_latest_empty_dir(self) -> None:
249249
self.assertEqual(
250250
log.output,
251251
[
252-
f"WARNING:torchtnt.framework.callbacks._checkpoint_utils:Input dirpath doesn't contain any subdirectories: {temp_dir}"
252+
f"WARNING:torchtnt.utils.checkpoint:Input dirpath doesn't contain any subdirectories: {temp_dir}"
253253
],
254254
)
255255
self.assertFalse(restored)

0 commit comments

Comments
 (0)