calculation matrix fix

OurPlanscape · Jan 8, 2025 · 4812dad · 4812dad
1 parent c9c65d0
commit 4812dad
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/planscape/impacts/models.py b/src/planscape/impacts/models.py
@@ -377,11 +377,10 @@ def get_baseline_only_impact_variables(cls) -> List:
 
     @classmethod
     def get_measurable_impact_variables(cls) -> List:
-        variables = set(cls.choices)
-        for variable in variables:
-            if variable in cls.get_baseline_only_impact_variables():
-                variables.remove(variable)
-        return list(variables)
+        variables = list(cls)
+        for baseline_variable in cls.get_baseline_only_impact_variables():
+            variables.remove(baseline_variable)
+        return variables
 
     @classmethod
     def get_datalayer(