Testing another fix

dataset.py

@@ -46,7 +46,9 @@ class DatasetWrapper(ClassificationDataset):
         return len(self.dataset)
 
     def __getitem__(self, idx):
-        path, label = self.dataset[idx]
+        item = self.dataset[idx]
+        print(item)
+        path, label = item["image"], item["label"]
         img = Image.open(path)
         if self.transforms:
             img = self.transforms(img)