Nol00 commited on
Commit
835d6ca
·
verified ·
1 Parent(s): 00f8114

Update src/model/crb_crs/recommender/movie_recommender.py

Browse files
src/model/crb_crs/recommender/movie_recommender.py CHANGED
@@ -534,10 +534,20 @@ class MovieRecommender(Recommender):
534
  Args:
535
  movie_id: Movie ID.
536
 
 
 
 
537
  Returns:
538
  Movie title.
539
  """
540
- return self.movie_mentions_df.loc[[int(movie_id)]]["title"].iloc[0]
 
 
 
 
 
 
 
541
 
542
  def replace_item_ids_with_recommendations(
543
  self,
 
534
  Args:
535
  movie_id: Movie ID.
536
 
537
+ Raises:
538
+ KeyError: if movie title cannot be found.
539
+
540
  Returns:
541
  Movie title.
542
  """
543
+ try:
544
+ title = self.movie_mentions_df.loc[[int(movie_id)]]["title"].iloc[
545
+ 0
546
+ ]
547
+ except KeyError:
548
+ title = ""
549
+ logging.error(f"Movie title not found for movie ID {movie_id}.")
550
+ return title
551
 
552
  def replace_item_ids_with_recommendations(
553
  self,