diff --git a/.travis.yml b/.travis.yml index f2bba7a041..6732f2cf3d 100644 --- a/.travis.yml +++ b/.travis.yml @@ -14,15 +14,8 @@ before_install: install: - conda create --yes -n gensim-test python=$TRAVIS_PYTHON_VERSION pip atlas numpy scipy - source activate gensim-test - - pip install pyemd - - pip install annoy - - pip install testfixtures - - pip install unittest2 - - pip install scikit-learn - - pip install tensorflow - - pip install keras - - pip install Morfessor==2.0.2a4 - python setup.py install + - pip install .[test] script: - python setup.py test - pip install flake8 diff --git a/setup.py b/setup.py index f1641fc59a..fa4b5caa8a 100644 --- a/setup.py +++ b/setup.py @@ -233,7 +233,8 @@ def finalize_options(self): 'scikit-learn', 'pyemd', 'annoy', - 'keras >= 2.0.4' + 'tensorflow >= 1.1.0', + 'keras >= 2.0.4', ] setup(