v0.2.1

Author: minimaxir

requirements.txt

@@ -1,2 +1,6 @@
-tensorflow
+pandas
+scikit-learn
+autopep8
+tqdm
+jinja2>=2.8
 pyyaml

setup.py

@@ -23,7 +23,7 @@
 setup(
     name='automl_gs',
     packages=['automl_gs'],  # this must be the same as the name above
-    version='0.2',
+    version='0.2.1',
     description='Provide an input CSV and a target field to predict, ' \
     'generate a model + code to run it.',
     long_description=long_description,
@@ -39,5 +39,5 @@
     },
     python_requires='>=3.5',
     include_package_data=True,
-    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8', 'tensorflow', 'pyyaml']
+    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8', 'pyyaml']
 )