Update recommenders/models/newsrec/models/base_model.py

Co-authored-by: Miguel Fierro <[email protected]>
recommenders-team · Jul 6, 2024 · 650ad36 · 650ad36
1 parent 2a9604e
commit 650ad36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recommenders/models/newsrec/models/base_model.py b/recommenders/models/newsrec/models/base_model.py
@@ -214,7 +214,7 @@ def fit(
             )
 
             for batch_data_input in tqdm_util:
-                if step_limit is not None and step_limit>=step_limit:
+                if step_limit is not None and step>=step_limit:
                     break
 
                 step_result = self.train(batch_data_input)