corrected init

2025-01-26 13:48:52 +01:00 · 2025-01-26 13:48:52 +01:00 · 00168af32d
parent 338ac5dee5
commit 00168af32d
1 changed files with 1 additions and 1 deletions
--- a/src/aiia/data/DataLoader.py
+++ b/src/aiia/data/DataLoader.py
@ -92,7 +92,7 @@ class AIIADataLoader(DataLoader):
                 seed=42,
                 column="file_path",
                 label_column=None):
-        super().__init__()
+        super().__init__(dataset)
        
        self.batch_size = batch_size
        self.val_split = val_split