Merge pull request #1475 from chenyuwuxin/master

FIX: fix compatibility issues for dataloader.dataset

recbole/data/dataloader/abstract_dataloader.py

@@ -22,6 +22,8 @@
 from recbole.utils import InputType, FeatureType, FeatureSource
 from recbole.data.transform import construct_transform
 
+start_iter = False
+
 
 class AbstractDataLoader(torch.utils.data.DataLoader):
     """:class:`AbstractDataLoader` is an abstract object which would return a batch of data which is loaded by
@@ -96,6 +98,19 @@ def collate_fn(self):
         """Collect the sampled index, and apply neg_sampling or other methods to get the final data."""
         raise NotImplementedError("Method [collate_fn] must be implemented.")
 
+    def __iter__(self):
+        global start_iter
+        start_iter = True
+        res = super().__iter__()
+        start_iter = False
+        return res
+
+    def __getattribute__(self, __name: str):
+        global start_iter
+        if not start_iter and __name == "dataset":
+            __name = "_dataset"
+        return super().__getattribute__(__name)
+
 
 class NegSampleDataLoader(AbstractDataLoader):
     """:class:`NegSampleDataLoader` is an abstract class which can sample negative examples by ratio.