[MNT] add type and length checking to dataset

abl/dataset/bridge_dataset.py

@@ -15,6 +15,11 @@ class BridgeDataset(Dataset):
         Y : List[Any]
             A list of objects representing the label.
         """
+        if (not isinstance(X, list)) or (not isinstance(Y, list)):
+            raise ValueError("X and Y should be of type list.")
+        if len(X) != len(Y):
+            raise ValueError("Length of X and Y must be equal.")
+        
         self.X = X
         self.Z = Z
         self.Y = Y

abl/dataset/classification_dataset.py

@@ -19,6 +19,11 @@ class ClassificationDataset(Dataset):
         transform : Callable[..., Any], optional
             A function/transform that takes in an object and returns a transformed version. Defaults to None.
         """
+        if (not isinstance(X, list)) or (not isinstance(Y, list)):
+            raise ValueError("X and Y should be of type list.")
+        if len(X) != len(Y):
+            raise ValueError("Length of X and Y must be equal.")
+        
         self.X = X
         self.Y = torch.LongTensor(Y)
         self.transform = transform

abl/dataset/regression_dataset.py

@@ -14,6 +14,11 @@ class RegressionDataset(Dataset):
         Y : List[Any]
             A list of objects representing the output data.
         """
+        if (not isinstance(X, list)) or (not isinstance(Y, list)):
+            raise ValueError("X and Y should be of type list.")
+        if len(X) != len(Y):
+            raise ValueError("Length of X and Y must be equal.")
+        
         self.X = X
         self.Y = Y