Fixed metadata initialization

luigibonati · Nov 14, 2024 · af5d238 · af5d238
1 parent d50c12b
commit af5d238
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mlcolvar/data/dataset.py b/mlcolvar/data/dataset.py
@@ -45,6 +45,10 @@ def __init__(self, dictionary: dict=None, feature_names = None, metadata: dict =
         if len(dictionary) == 0:
             raise ValueError("Empty datasets are not supported")
 
+        # initialize metadata as dict
+        if metadata is None:
+            metadata = {}
+
         # convert to torch.Tensors
         for key, val in dictionary.items():
             if not isinstance(val, torch.Tensor):