From 181549bd17c11004e51410e83c53238e5454618b Mon Sep 17 00:00:00 2001
From: Rohan Anand <thededlier@gmail.com>
Date: Sat, 23 May 2020 18:31:47 +0530
Subject: [PATCH] initial pypi release

---
 auto_surprise/trainer.py |  1 +
 examples/ml_100k.py      |  6 ++----
 setup.cfg                |  2 ++
 setup.py                 | 17 +++++++++++------
 4 files changed, 16 insertions(+), 10 deletions(-)
 create mode 100644 setup.cfg

diff --git a/auto_surprise/trainer.py b/auto_surprise/trainer.py
index 227ffa1..22fad58 100644
--- a/auto_surprise/trainer.py
+++ b/auto_surprise/trainer.py
@@ -88,6 +88,7 @@ def start_with_limits(self, max_evals, time_limit, tasks):
                     }
 
             except Exception as e:
+                print("Exception for algo ", self._algo_name)
                 print(traceback.format_exc())
 
                 if self._debug:
diff --git a/examples/ml_100k.py b/examples/ml_100k.py
index 8e6d76e..49a2bcc 100644
--- a/examples/ml_100k.py
+++ b/examples/ml_100k.py
@@ -2,13 +2,11 @@
 import datetime
 import os
 import sys
-
-sys.path.insert(1, './')
-
+from surprise import Dataset
 from auto_surprise.engine import Engine
 
 if __name__ == '__main__':
-    
+
     data = Dataset.load_builtin('ml-100k')
 
     # Run auto surprise
diff --git a/setup.cfg b/setup.cfg
new file mode 100644
index 0000000..0d9f081
--- /dev/null
+++ b/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+description-file = README.md
diff --git a/setup.py b/setup.py
index a078015..01817ff 100644
--- a/setup.py
+++ b/setup.py
@@ -1,21 +1,26 @@
 import setuptools
 
-# default required packages
-install_requires = ['hyperopt', 'scikit-surprise']
+install_requires = [
+    "hyperopt",
+    "lightgbm",
+    "numpy",
+    "scikit-learn==0.22.0",
+    "scikit-surprise"
+]
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setuptools.setup(
     name="auto-surprise",
-    packages="auto-surprise",
-    version="0.0.1",
+    version="0.1.0",
     author="Rohan Anand",
     author_email="anandr@tcd.ie",
-    description="A python package that automates machine learning for the recommender system library Surprise",
+    description="A python package that automates algorithm selection and hyperparameter tuning for the recommender system library Surprise",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/thededlier/auto-surprise",
+    url="https://github.com/BeelGroup/Auto-Surprise",
+    download_url="https://github.com/BeelGroup/Auto-Surprise/archive/v0.1.0.tar.gz",
     packages=setuptools.find_packages(),
     classifiers=[
         "Programming Language :: Python :: 3",