fix ihheritance

Signed-off-by: novahow <[email protected]>
flyteorg · Dec 13, 2024 · 340365e · 340365e
1 parent d3fc20d
commit 340365e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flytekit/core/checkpointer.py b/flytekit/core/checkpointer.py
@@ -165,7 +165,7 @@ def write(self, b: bytes):
         self.save(f)
 
 
-class TorchAsyncCheckpoint(Checkpoint):
+class TorchAsyncCheckpoint(SyncCheckpoint):
     """
     This class is NOT THREAD-SAFE!
     Sync Checkpoint, will synchronously checkpoint a user given file or folder.