fix: return None if loading failed to avoid using var data before ass…

…ignment;
WenjieDu · Jul 4, 2024 · 4200c94 · 4200c94
1 parent 8993137
commit 4200c94
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pypots/data/saving/pickle.py b/pypots/data/saving/pickle.py
@@ -59,4 +59,5 @@ def pickle_load(path: str) -> object:
             data = pickle.load(f)
     except Exception as e:
         logger.error(f"❌ Loading data failed. Operation aborted. See info below:\n{e}")
+        return None
     return data