Changed assert to ValueError

Signed-off-by: David Davó <[email protected]>
recommenders-team · Sep 9, 2024 · 84497f2 · 84497f2
1 parent b049091
commit 84497f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/recommenders/models/deeprec/DataModel/ImplicitCF.py b/recommenders/models/deeprec/DataModel/ImplicitCF.py
@@ -206,7 +206,8 @@ def train_loader(self, batch_size):
         """
 
         def sample_neg(x):
-            assert len(x) < self.n_items, "A user has voted in every item. Can't find a negative sample"
+            if len(x) >= self.n_items:
+                raise ValueError("A user has voted in every item. Can't find a negative sample.")
             while True:
                 neg_id = random.randint(0, self.n_items - 1)
                 if neg_id not in x: