refact(#7): refact lowercase title

fedepacher · Jun 11, 2023 · 2ad0ed9 · 2ad0ed9
1 parent 6e53471
commit 2ad0ed9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recommendation_system.py b/recommendation_system.py
@@ -29,7 +29,7 @@ def recommendation(self, title=''):
         Returns:
             str: Dictionary with the movies recommended.
         """
-        movie_index = self._df_movies[self._df_movies['title'].str.lower() == title].index.values
+        movie_index = self._df_movies[self._df_movies['title'].str.lower() == title.lower()].index.values
         if len(movie_index) > 0:
             similarity_scores = cosine_similarity(self._tfidf_matrix[movie_index],
                                                   self._tfidf_matrix)