Permalink
Browse files

No more warnings in hyperbolic unit tests

  • Loading branch information...
ninamiolane committed Nov 16, 2018
1 parent d76a1aa commit 8a9f9374c7dfddbd21d510a815228c730e6946f4
Showing with 5 additions and 2 deletions.
  1. +2 −1 geomstats/__about__.py
  2. +2 −1 requirements.txt
  3. +1 −0 tests/test_hyperbolic_space_tensorflow.py
@@ -5,14 +5,15 @@

install_requires = [
'autograd',
'h5py==2.8.0',
'matplotlib',
'numpy>=1.14.1',
'scipy',
]

extras_require = {
'test': ['codecov', 'coverage', 'nose2'],
'tf': ['tensorflow>=1.8'],
'tf': ['tensorflow>=1.12'],
'torch': ['torch==0.4.0'],
}
extras_require['all'] = list(chain(*extras_require.values()))
@@ -1,9 +1,10 @@
autograd
codecov
coverage
h5py==2.8.0
matplotlib
nose2
numpy>=1.14.1
scipy
tensorflow>=1.8
tensorflow>=1.12
torch==0.4.0
@@ -34,6 +34,7 @@ def setUp(self):
@classmethod
def setUpClass(cls):
os.environ['GEOMSTATS_BACKEND'] = 'tensorflow'
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
importlib.reload(gs)

@classmethod

0 comments on commit 8a9f937

Please sign in to comment.