diff --git a/README.md b/README.md index f2f126e..d884cd1 100644 --- a/README.md +++ b/README.md @@ -99,6 +99,7 @@ Type: type ## Contributors - Dileep Kishore +- Blessing Ibe ## Credits diff --git a/movie_recommender/movie_recommender.py b/movie_recommender/movie_recommender.py index 8e3bcaf..1b1f989 100644 --- a/movie_recommender/movie_recommender.py +++ b/movie_recommender/movie_recommender.py @@ -7,6 +7,7 @@ """ from typing import List +from .utils import save, load, search import warnings warnings.filterwarnings("ignore", message="numpy.dtype size changed") warnings.filterwarnings("ignore", message="numpy.ufunc size changed") @@ -70,5 +71,8 @@ def movie_recommender( recommender = KeywordSimilarity(metadata, credits, keywords) elif method == "UserSimilarity": recommender = UserSimilarity(metadata, ratings, links) + else: + raise ValueError() print("Returning recommendations") + save(recommender) return recommender(movie, nhits=nhits)