make BasicModel consider all constraint groups

oemof · Oct 20, 2023 · f73e7db · f73e7db
1 parent 119e9cb
commit f73e7db
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/src/oemof/solph/_models.py b/src/oemof/solph/_models.py
@@ -132,12 +132,21 @@ def __init__(self, energysystem, **kwargs):
             "constraint_groups", []
         )
 
-        self._constraint_groups += [
-            i
-            for i in self.es.groups
-            if hasattr(i, "CONSTRAINT_GROUP")
-            and i not in self._constraint_groups
-        ]
+        if self.is_cellular:
+            for es in energysystem:
+                self._constraint_groups += [
+                    i
+                    for i in es.groups
+                    if hasattr(i, "CONSTRAINT_GROUP")
+                    and i not in self._constraint_groups
+                ]
+        else:
+            self._constraint_groups += [
+                i
+                for i in self.es.groups
+                if hasattr(i, "CONSTRAINT_GROUP")
+                and i not in self._constraint_groups
+            ]
 
         self.flows = self.es.flows()
         if self.is_cellular: