Fix Tensorflow 2.0 error (#119)

asreview · Oct 16, 2019 · b58ac90 · b58ac90
1 parent 709487c
commit b58ac90
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -7,7 +7,7 @@ notifications:
   email: false
 
 install:
-    - pip install tensorflow==1.13.2
+    - pip install --upgrade setuptools>=41.0.0
     - pip install .
 
 script:

diff --git a/asreview/__main__.py b/asreview/__main__.py
@@ -13,7 +13,10 @@
 
 import numpy as np  # noqa
 import tensorflow as tf # noqa
-tf.logging.set_verbosity(tf.logging.ERROR)
+try:
+    tf.logging.set_verbosity(tf.logging.ERROR)
+except AttributeError:
+    logging.getLogger("tensorflow").setLevel(logging.ERROR)
 
 from asreview import __version__  # noqa
 from asreview.review import review_oracle, review_simulate  # noqa