Fix

iai-group · Sep 11, 2024 · 2886b2b · 2886b2b
1 parent f0ece02
commit 2886b2b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/model/crb_crs/recommender/movie_recommender.py b/src/model/crb_crs/recommender/movie_recommender.py
@@ -537,7 +537,14 @@ def get_movie_title(self, movie_id: str) -> str:
         Returns:
             Movie title.
         """
-        return self.movie_mentions_df.loc[[int(movie_id)]]["title"].iloc[0]
+        try:
+            title = self.movie_mentions_df.loc[[int(movie_id)]]["title"].iloc[
+                0
+            ]
+        except KeyError:
+            title = ""
+            logging.error(f"Movie title not found for movie ID {movie_id}.")
+        return title
 
     def replace_item_ids_with_recommendations(
         self,