rename torchsnapshot_saver_types to checkpointer_types (#625)

Summary: Pull Request resolved: #625 Renames `torchsnapshot_saver_types.py` to `checkpointer_types.py` as some of the dataclasses can be reused in other checkpointing callbacks (and helps create a unified api for restoring checkpoints through the use of `RestoreOptions` across all checkpointers Reviewed By: anshulverma Differential Revision: D51408793 fbshipit-source-id: 958471750076cd6cf09b99babcf86d85e439a6d5
pytorch · Nov 20, 2023 · 03b8ec6 · 03b8ec6
1 parent 223eeee
commit 03b8ec6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/...rk/callbacks/torchsnapshot_saver_types.py → ...framework/callbacks/checkpointer_types.py b/...rk/callbacks/torchsnapshot_saver_types.py → ...framework/callbacks/checkpointer_types.py
diff --git a/torchtnt/framework/callbacks/torchsnapshot_saver.py b/torchtnt/framework/callbacks/torchsnapshot_saver.py
@@ -26,10 +26,7 @@
 from pyre_extensions import none_throws
 
 from torchtnt.framework.callback import Callback
-from torchtnt.framework.callbacks.torchsnapshot_saver_types import (
-    KnobOptions,
-    RestoreOptions,
-)
+from torchtnt.framework.callbacks.checkpointer_types import KnobOptions, RestoreOptions
 from torchtnt.framework.state import EntryPoint, State
 from torchtnt.framework.unit import (
     AppStateMixin,