fix typo on logger warning in general_dataloader.py

RUCAIBox · Jan 21, 2024 · e898aaa · e898aaa
1 parent 4121d5c
commit e898aaa
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)