From 6d753cd4659c95b431551b122fa153e5814c4fdf Mon Sep 17 00:00:00 2001 From: Pierre-Louis Barbarant Date: Mon, 15 Jul 2024 18:32:07 +0200 Subject: [PATCH] Add iter name --- src/fugw/mappings/sparse_barycenter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fugw/mappings/sparse_barycenter.py b/src/fugw/mappings/sparse_barycenter.py index dd2521e..1bd8383 100644 --- a/src/fugw/mappings/sparse_barycenter.py +++ b/src/fugw/mappings/sparse_barycenter.py @@ -254,7 +254,7 @@ def fit( mask = None losses_each_bar_step = [] - for _ in range(nits_barycenter): + for idx in range(nits_barycenter): # Transport all elements plans, losses = self.compute_all_ot_plans( plans,