Add iter name

alexisthual · Jul 15, 2024 · 6d753cd · 6d753cd
1 parent 4e525da
commit 6d753cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,