Mention that a numpy array must be passed to doc arg

pieronima · Jul 11, 2019 · 9d2d22a · 9d2d22a
1 parent bcaf7ca
commit 9d2d22a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/libraries/keras.rst b/docs/source/libraries/keras.rst
@@ -34,7 +34,7 @@ Important arguments to :func:`eli5.explain_prediction` for ``Model`` and ``Seque
 
 * ``doc`` is an image as a tensor that can be inputted to the model.
 
-    - The type is a ``numpy.ndarray``.
+    - The tensor must be an instance of ``numpy.ndarray``.
 
     - Usually the tensor has the format `(batch, dims, ..., channels)` (channels last format, `dims=(height, width)`, `batch=1`, one image), i.e. `BHWC`.
 

diff --git a/eli5/keras/explain_prediction.py b/eli5/keras/explain_prediction.py
@@ -47,7 +47,7 @@ def explain_prediction_keras(estimator, # type: Model
         An input image as a tensor to ``estimator``, 
         from which prediction will be done and explained.
 
-        For example a ``numpy.ndarray``.
+        Currently only numpy arrays are supported.
 
         The tensor must be of suitable shape for the ``estimator``.