From b73712a3d5e5fcf4960eb602f7f043b975ae256a Mon Sep 17 00:00:00 2001
From: Iglesia Dolci <daia.dolci@gmail.com>
Date: Tue, 12 Nov 2024 22:31:55 +0000
Subject: [PATCH] DiskCheckpointingManager

---
 firedrake/adjoint_utils/checkpointing.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/firedrake/adjoint_utils/checkpointing.py b/firedrake/adjoint_utils/checkpointing.py
index e6ec1c6816..39ec4cc351 100644
--- 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