DiskCheckpointingManager

firedrakeproject · Nov 12, 2024 · b73712a · b73712a
1 parent d6f960d
commit b73712a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/firedrake/adjoint_utils/checkpointing.py b/firedrake/adjoint_utils/checkpointing.py
@@ -1,7 +1,7 @@
 """A module providing support for disk checkpointing of the adjoint tape."""
 from pyadjoint import get_working_tape, OverloadedType
 from pyadjoint.tape import TapePackageData
-from pyadjoint.checkpointing import ManageDiskCheckpointing
+from pyadjoint.checkpointing import DiskCheckpointingManager
 from pyop2.mpi import COMM_WORLD
 import tempfile
 import os
@@ -351,7 +351,7 @@ def _ad_restore_at_checkpoint(self, checkpoint):
         return checkpoint
 
 
-class AdjointDiskCheckpointing(ManageDiskCheckpointing):
+class AdjointDiskCheckpointing(DiskCheckpointingManager):
     """Manager for the disk checkpointing process.
 
     Parameters