Add methods required by interface to PyGrainCheckpointHandler.

PiperOrigin-RevId: 575646928
google · Oct 22, 2023 · 8f20bf6 · 8f20bf6
1 parent 75f6194
commit 8f20bf6
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/grain/_src/python/checkpoint_handlers.py b/grain/_src/python/checkpoint_handlers.py
@@ -52,3 +52,15 @@ def restore(
   def structure(self, directory: epath.Path) -> Any:
     del directory
     return None
+
+  # Required by interface.
+
+  def metadata(self, directory: epath.Path) -> Optional[Any]:
+    del directory
+    return None
+
+  def finalize(self, directory: epath.Path):
+    pass
+
+  def close(self):
+    pass