From a53b5e193ddad9ba30148669b340cf1241827fba Mon Sep 17 00:00:00 2001
From: Lennart Philipp <lennart-philipp@outlook.de>
Date: Mon, 22 Apr 2024 10:43:12 +0200
Subject: [PATCH] fixes a bug that occurs when trying to open the downloaded
 model weights

---
 brainles_aurora/inferer/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/brainles_aurora/inferer/model.py b/brainles_aurora/inferer/model.py
index 71b69a0..97b6154 100644
--- a/brainles_aurora/inferer/model.py
+++ b/brainles_aurora/inferer/model.py
@@ -85,7 +85,7 @@ def _load_model(self, num_input_modalities: int) -> torch.nn.Module:
         # load weights
         weights_path = os.path.join(
             self.model_weights_folder,
-            f"{self.inference_mode}_{self.config.model_selection}.tar",
+            f"{self.inference_mode.value}_{self.config.model_selection.value}.tar",
         )
         if not os.path.exists(weights_path):
             raise NotImplementedError(