Issue yell#8 Fix builds

SrMouraSilva · Mar 6, 2019 · 24f0411 · 24f0411
1 parent 2923992
commit 24f0411
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 12 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -18,12 +18,10 @@ install:
   - pip install -U setuptools
   - pip uninstall --yes pytest
   - python setup_cpu.py develop
-  - make test
-  - codecov
 
 script:
   - make test
-  - #make test-docs
+  - codecov
 
 after_success:
   - bash <(curl -s https://codecov.io/bash)
diff --git a/setup.py b/setup.py
@@ -4,16 +4,16 @@
 setup(name='boltzmann_machines',
       version='0.1',
       url='https://github.com/monsta-hd/boltzmann-machines',
-      install_requires=['tensorflow==1.13.*',
-                        'scipy>=0.17', 
-                        'keras>=2.0.8', 
-                        'matplotlib>=1.5.3',
+      install_requires=['tensorflow-gpu>=1.13.*,<2.0.0',
+                        'scipy>=0.17',
+                        'keras>=2.0.8',
+                        'matplotlib>=1.5.3,<3.0.0',
                         'numpy>=1.16.0',
                         'scikit-learn>=0.19.0',
-                        'seaborn>=0.8.1', 
+                        'seaborn>=0.8.1',
                         'tqdm>=4.14.0',
                         # Tests
-                        'pytest==4.3.*',
+                        'pytest>=4.3.*',
                         'pytest-cov>=2.6.1',
                         'nose>=1.3.4',
                         'nose-exclude>=0.5.0',

diff --git a/setup_cpu.py b/setup_cpu.py
@@ -4,16 +4,16 @@
 setup(name='boltzmann_machines',
       version='0.1',
       url='https://github.com/monsta-hd/boltzmann-machines',
-      install_requires=['tensorflow==1.13.*',
+      install_requires=['tensorflow>=1.13.*,<2.0.0',
                         'scipy>=0.17',
                         'keras>=2.0.8',
-                        'matplotlib>=1.5.3',
+                        'matplotlib>=1.5.3,<3.0.0',
                         'numpy>=1.16.0',
                         'scikit-learn>=0.19.0',
                         'seaborn>=0.8.1',
                         'tqdm>=4.14.0',
                         # Tests
-                        'pytest==4.3.*',
+                        'pytest>=4.3.*',
                         'pytest-cov>=2.6.1',
                         'nose>=1.3.4',
                         'nose-exclude>=0.5.0',