Merge pull request #1976 from al-difonzo/patch-1

Fix typo on logger warning in general_dataloader.py
RUCAIBox · Mar 2, 2024 · 48ce443 · 48ce443
2 parents f95fde5 + e898aaa
commit 48ce443
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recbole/data/dataloader/general_dataloader.py b/recbole/data/dataloader/general_dataloader.py
@@ -113,7 +113,7 @@ def __init__(self, config, dataset, sampler, shuffle=False):
         else:
             self.sample_size = len(dataset)
         if shuffle:
-            self.logger.warnning("NegSampleEvalDataLoader can't shuffle")
+            self.logger.warning("NegSampleEvalDataLoader can't shuffle")
             shuffle = False
         super().__init__(config, dataset, sampler, shuffle=shuffle)
 
@@ -227,7 +227,7 @@ def __init__(self, config, dataset, sampler, shuffle=False):
 
         self.sample_size = len(self.user_df) if not self.is_sequential else len(dataset)
         if shuffle:
-            self.logger.warnning("FullSortEvalDataLoader can't shuffle")
+            self.logger.warning("FullSortEvalDataLoader can't shuffle")
             shuffle = False
         super().__init__(config, dataset, sampler, shuffle=shuffle)