Merge pull request #16 from krzynio/master

Pyyaml warning fix and dependencies
minimaxir · Apr 5, 2019 · 82ae6d0 · 82ae6d0
2 parents aba6dbf + 779bcea
commit 82ae6d0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,2 @@
+tensorflow
+pyyaml
diff --git a/setup.py b/setup.py
@@ -39,5 +39,5 @@
     },
     python_requires='>=3.5',
     include_package_data=True,
-    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8']
+    install_requires=['pandas', 'scikit-learn', 'autopep8', 'tqdm', 'jinja2>=2.8', 'tensorflow', 'pyyaml']
 )