From ecd6d756487a421abb865feb1d6b6bd6db2ffa73 Mon Sep 17 00:00:00 2001 From: myui Date: Sun, 2 Feb 2025 09:41:12 +0900 Subject: [PATCH] Fixed a CI error --- rtrec/models/base.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/rtrec/models/base.py b/rtrec/models/base.py index 1104920..d9561eb 100644 --- a/rtrec/models/base.py +++ b/rtrec/models/base.py @@ -182,11 +182,13 @@ def recommend_batch(self, users: List[Any], candidate_items: Optional[List[Any]] :param filter_interacted: Whether to filter out items the user has already interacted with :return: List of top-K items recommended for each user """ - user_ids = [ - None if self.user_ids.pass_through and (user_id := self.user_ids.get_id(user)) > self.interactions.max_user_id - else user_id - for user in users - ] + user_ids = [] + for user in users: + uid = self.user_ids.get_id(user) + if self.user_ids.pass_through and uid > self.interactions.max_user_id: + user_ids.append(None) + else: + user_ids.append(uid) candidate_item_ids = None if candidate_items is not None: candidate_item_ids = [