Merge pull request #1879 from zhengbw0324/master

FIX: Update abstract_dataloader.py
RUCAIBox · Sep 26, 2023 · 23069f6 · 23069f6
2 parents 4d7822f + c5be7ef
commit 23069f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recbole/data/dataloader/abstract_dataloader.py b/recbole/data/dataloader/abstract_dataloader.py
@@ -136,8 +136,8 @@ def _set_neg_sample_args(self, config, dataset, dl_format, neg_sample_args):
         self.neg_sample_args = neg_sample_args
         self.times = 1
         if (
-            self.neg_sample_args["distribution"] == "uniform"
-            or self.neg_sample_args["distribution"] == "popularity"
+            (self.neg_sample_args["distribution"] == "uniform"
+            or self.neg_sample_args["distribution"] == "popularity")
             and self.neg_sample_args["sample_num"] != "none"
         ):
             self.neg_sample_num = self.neg_sample_args["sample_num"]