summaryrefslogtreecommitdiff
path: root/ot/plot.py
diff options
context:
space:
mode:
authorAlexandre Gramfort <alexandre.gramfort@m4x.org>2019-07-09 17:20:02 +0200
committerAlexandre Gramfort <alexandre.gramfort@m4x.org>2019-07-09 17:20:02 +0200
commit06fab4c1e5efbe79f91589917fba01c3fb300a87 (patch)
treeef1c832df0e3e4cb8aee044ae98751c9fac608aa /ot/plot.py
parentb6fb14861accd20a323bfc5ef96c20883e4f6ce1 (diff)
more
Diffstat (limited to 'ot/plot.py')
-rw-r--r--ot/plot.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/ot/plot.py b/ot/plot.py
index a409d4a..f403e98 100644
--- a/ot/plot.py
+++ b/ot/plot.py
@@ -26,11 +26,11 @@ def plot1D_mat(a, b, M, title=''):
Parameters
----------
- a : np.array, shape (na,)
+ a : ndarray, shape (na,)
Source distribution
- b : np.array, shape (nb,)
+ b : ndarray, shape (nb,)
Target distribution
- M : np.array, shape (na,nb)
+ M : ndarray, shape (na, nb)
Matrix to plot
"""
na, nb = M.shape