Fix Reader import in main.py

NicolasHug · Jul 21, 2019 · 89fb7a2 · 89fb7a2
1 parent 77f7d41
commit 89fb7a2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/surprise/__main__.py b/surprise/__main__.py
@@ -22,6 +22,7 @@
 from surprise.prediction_algorithms import CoClustering
 import surprise.dataset as dataset
 from surprise.dataset import Dataset
+from surprise.reader import Reader
 from surprise.builtin_datasets import get_dataset_dir
 from surprise.model_selection import cross_validate
 from surprise.model_selection import KFold