From 71280106acf09a67b54600eb1f993284e127ed35 Mon Sep 17 00:00:00 2001
From: David Kuegler <david.kuegler@dzne.de>
Date: Mon, 9 Sep 2024 16:46:42 +0200
Subject: [PATCH] Restore CerebNet checkpoint functions

Restore FastSurfer checkpoint functions inherited by CerebNet via import.
Those functions are used in downstream modules.
---
 CerebNet/utils/checkpoint.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/CerebNet/utils/checkpoint.py b/CerebNet/utils/checkpoint.py
index 2c83469d..38726ba2 100644
--- a/CerebNet/utils/checkpoint.py
+++ b/CerebNet/utils/checkpoint.py
@@ -19,9 +19,26 @@
 if TYPE_CHECKING:
     import yacs
 
+from FastSurferCNN.utils.checkpoint import (
+    load_from_checkpoint,
+    create_checkpoint_dir,
+    get_checkpoint,
+    get_checkpoint_path,
+    save_checkpoint,
+)
 from FastSurferCNN.utils import logging
 from FastSurferCNN.utils.parser_defaults import FASTSURFER_ROOT
 
+__all__ = [
+    "create_checkpoint_dir",
+    "get_checkpoint",
+    "get_checkpoint_path",
+    "is_checkpoint_epoch",
+    "load_from_checkpoint",
+    "save_checkpoint",
+    "YAML_DEFAULT",
+]
+
 # DEFAULTS
 YAML_DEFAULT = FASTSURFER_ROOT / "CerebNet/config/checkpoint_paths.yaml"