improved the pylint score

adipai · Oct 16, 2023 · 4cae9b1 · 4cae9b1
1 parent c681a3f
commit 4cae9b1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/prediction_scripts/item_based.py b/src/prediction_scripts/item_based.py
@@ -49,4 +49,6 @@ def recommend_for_new_user(user_rating):
         by="recommended", ascending=False, inplace=True
     )
 
-    return list(join_movies_and_recommendations["title"][:201]), list(join_movies_and_recommendations["genres"][:201]), list(join_movies_and_recommendations["imdb_id"][:201])
+    return list(join_movies_and_recommendations["title"][:201]), \
+        list(join_movies_and_recommendations["genres"][:201]), \
+            list(join_movies_and_recommendations["imdb_id"][:201])