pypi

python-package/requirements.txt

@@ -3,8 +3,8 @@ matplotlib>=2.0.2
 hyperopt==0.1
 Keras==2.1.2
 pandas==0.21.0
+joblib==0.11
 scikit-learn==0.19.0
-joblib==0.13.2
 Theano==1.0.1
 lightgbm==2.0.11
 networkx==1.11

python-package/setup.py

@@ -2,9 +2,10 @@
 # -*- coding: utf-8 -*-
 
 import pip
+from pip._internal import main
 from setuptools import setup
 from setuptools.command.install import install
-from mlbox.__init__ import __version__
+
 
 with open('requirements.txt', 'rt') as fh:
     requirements = fh.read().splitlines()
@@ -16,8 +17,12 @@ def run(self):
         failed = []
 
         for req in requirements:
-            if pip.main(["install", req]) == 1:
-                failed.append(req)
+            try:
+                if main(["install", req]) == 1:
+                    failed.append(req)
+            except:
+                if pip.main(["install", req]) == 1:
+                    failed.append(req)
 
         if len(failed) > 0:
             print("")
@@ -36,7 +41,7 @@ def run(self):
 
 setup(
     name='mlbox',
-    version=__version__,
+    version="0.6.1",
     description="A powerful Automated Machine Learning python library.",
     long_description=readme,
     author="Axel ARONIO DE ROMBLAY",