From b2b8f6cbe6be46d1ae873ff279d256f63fa991f1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20PINZ=C3=B3N?= Date: Fri, 17 Oct 2025 17:22:59 +0200 Subject: [PATCH] Update docstring to include 'a' parameter Added parameter description for 'a' in the docstring. --- ot/stochastic.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ot/stochastic.py b/ot/stochastic.py index 29a8783db..25b08c6ab 100644 --- a/ot/stochastic.py +++ b/ot/stochastic.py @@ -178,6 +178,8 @@ def averaged_sgd_entropic_transport( Parameters ---------- + a : ndarray, shape (ns,) + source measure b : ndarray, shape (nt,) target measure M : ndarray, shape (ns, nt)