MAINT: Adapt to numbering of heterogeneous labels.

pmgbergen · Nov 22, 2024 · 41f0584 · 41f0584
1 parent cb3811f
commit 41f0584
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/darsia/presets/analysis/multichromatictracer.py b/src/darsia/presets/analysis/multichromatictracer.py
@@ -163,9 +163,8 @@ def calibrate(self, colors, concentrations) -> None:
             concentrations (list): list of concentrations
 
         """
-        for i, _ in enumerate(darsia.Masks(self.labels)):
-
-            self.model[0][i].update(supports=colors[i], values=concentrations[i])
+        for i, (_, label) in enumerate(darsia.Masks(self.labels, return_label=True)):
+            self.model[0][label].update(supports=colors[i], values=concentrations[i])
 
     def calibrate_from_image(
         self,