From tensorflow==2.0.0 to tensorflow==2.1.0 (#111)

* From tensorflow==2.0.0 to tensorflow==2.1.0

* Use light cpu version of TF

* Why is it failing now?

docs/source/overview.rst

@@ -16,7 +16,4 @@ Tensorflow Compatibility
 tf-explain is compatible with Tensorflow 2. It is not declared as a dependency
 to let you choose between CPU and GPU versions. Additionally to the previous install,
 run::
-    # For CPU version
-    pip install tensorflow==2.0.0
-    # For GPU version
-    pip install tensorflow-gpu==2.0.0
+    pip install tensorflow==2.1.0

tests/core/test_integrated_gradients.py

@@ -35,6 +35,6 @@ def test_get_integrated_gradients(random_data, convolutional_model):
         images, convolutional_model, 0, n_steps=n_steps
     )
 
-    expected_output_shape = (images.shape[0] / n_steps, *images.shape[1:])
+    expected_output_shape = (int(images.shape[0] / n_steps), *images.shape[1:])
 
     assert gradients.shape == expected_output_shape

tests/core/test_smoothgrad.py

@@ -34,6 +34,6 @@ def test_get_averaged_gradients(random_data, convolutional_model):
         images, convolutional_model, 0, num_samples=num_samples
     )
 
-    expected_output_shape = (images.shape[0] / num_samples, *images.shape[1:])
+    expected_output_shape = (int(images.shape[0] / num_samples), *images.shape[1:])
 
     assert gradients.shape == expected_output_shape

tox.ini

@@ -10,5 +10,5 @@ deps =
     pytest-cov
     pytest-mock
 commands =
-    pip install tensorflow==2.0.0
+    pip install tensorflow-cpu==2.1.0
     pytest tests/ -vv --cov-report term-missing --no-cov-on-fail --cov-fail-under 85 --cov=tf_explain/