Update dataset.py

Author: MaxSchambach

mambular/data_utils/dataset.py

@@ -24,11 +24,14 @@ def __init__(
         labels=None,
         regression=True,
     ):
+        assert cat_features_list or num_features_list
+        
         self.cat_features_list = cat_features_list  # Categorical features tensors
         self.num_features_list = num_features_list  # Numerical features tensors
         self.embeddings_list = embeddings_list  # Embeddings tensors (optional)
         self.regression = regression
 
+
         if labels is not None:
             if not self.regression:
                 self.num_classes = len(np.unique(labels))