From 650ad365748b3be297c4a9a14d6c21c2b1c63034 Mon Sep 17 00:00:00 2001 From: Sumana Sree <110307215+sumana-2705@users.noreply.github.com> Date: Sat, 6 Jul 2024 09:07:57 +0530 Subject: [PATCH] Update recommenders/models/newsrec/models/base_model.py Co-authored-by: Miguel Fierro <3491412+miguelgfierro@users.noreply.github.com> --- recommenders/models/newsrec/models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recommenders/models/newsrec/models/base_model.py b/recommenders/models/newsrec/models/base_model.py index c19679c95..344ea8f7c 100644 --- 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)