changed parameter name to step_limit from steps

Signed-off-by: sumana sree <[email protected]>
recommenders-team · Jul 5, 2024 · 2a9604e · 2a9604e
1 parent 7c9a693
commit 2a9604e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/smoke/recommenders/recommender/test_newsrec_model.py b/tests/smoke/recommenders/recommender/test_newsrec_model.py
@@ -62,7 +62,7 @@ def test_model_nrms(mind_resource_path):
     assert model.run_eval(valid_news_file, valid_behaviors_file) is not None
     assert isinstance(
         model.fit(
-            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,steps=1
+            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,step_limit=1
         ),
         BaseModel,
     )
@@ -115,7 +115,7 @@ def test_model_naml(mind_resource_path):
     assert model.run_eval(valid_news_file, valid_behaviors_file) is not None
     assert isinstance(
         model.fit(
-            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,steps=1
+            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,step_limit=1
         ),
         BaseModel,
     )
@@ -166,7 +166,7 @@ def test_model_lstur(mind_resource_path):
     assert model.run_eval(valid_news_file, valid_behaviors_file) is not None
     assert isinstance(
         model.fit(
-            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,steps=1
+            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,step_limit=1
         ),
         BaseModel,
     )
@@ -217,7 +217,7 @@ def test_model_npa(mind_resource_path):
     assert model.run_eval(valid_news_file, valid_behaviors_file) is not None
     assert isinstance(
         model.fit(
-            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,steps=1
+            train_news_file, train_behaviors_file, valid_news_file, valid_behaviors_file,step_limit=1
         ),
         BaseModel,
     )