diff --git a/recbole/data/dataloader/abstract_dataloader.py b/recbole/data/dataloader/abstract_dataloader.py index 59f6398fd..b97b33174 100644 --- a/recbole/data/dataloader/abstract_dataloader.py +++ b/recbole/data/dataloader/abstract_dataloader.py @@ -136,10 +136,9 @@ 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") - and self.neg_sample_args["sample_num"] != "none" - ): + 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"] if self.dl_format == InputType.POINTWISE: