Merge pull request #1651 from ywkim92/patch-1

fix: Change formats of user_ids and item_ids arg.

Author: miguelgfierro

recommenders/models/lightfm/lightfm_utils.py

@@ -253,8 +253,8 @@ def prepare_all_predictions(
 
     all_predictions["prediction"] = all_predictions.apply(
         lambda x: model.predict(
-            user_ids=int(x["uid"]),
-            item_ids=[x["iid"]],
+            user_ids=np.array([x["uid"]], dtype=np.int32),
+            item_ids=np.array([x["iid"]], dtype=np.int32),
             user_features=user_features,
             item_features=item_features,
             num_threads=num_threads,