Update dataset.py

MaxSchambach · web-flow · commit 7ade7843e4bc · 2025-07-26T17:24:34.000+02:00
diff --git a/mambular/data_utils/dataset.py b/mambular/data_utils/dataset.py
@@ -44,7 +44,7 @@ def __init__(
             self.labels = None  # No labels in prediction mode
 
     def __len__(self):
-        _feats = self.num_features_list if self.num_features_list else cat_features_list
+        _feats = self.num_features_list if self.num_features_list else self.cat_features_list
         return len(_feats[0])
 
     def __getitem__(self, idx):
