Fix ignore attribute for disappear costs

funkelab · Jan 22, 2024 · 3ec2916 · 3ec2916
1 parent c160b19
commit 3ec2916
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/motile/costs/disappear.py b/motile/costs/disappear.py
@@ -24,14 +24,14 @@ class Disappear(Costs):
             at no cost.
     """
 
-    def __init__(self, constant: float) -> None:
+    def __init__(self, constant: float, ignore_attribute: str | None = None) -> None:
         self.constant = Weight(constant)
         self.ignore_attribute = ignore_attribute
 
     def apply(self, solver: Solver) -> None:
         disappear_indicators = solver.get_variables(NodeDisappear)
 
-        for index in disappear_indicators.values():
+        for node, index in disappear_indicators.items():
             if self.ignore_attribute is not None:
                 if solver.graph.nodes[node].get(self.ignore_attribute, False):
                     continue