clarify exceptions in from_microgrid

Total-RD · Dec 17, 2022 · c1f1d55 · c1f1d55
1 parent 21fc24f
commit c1f1d55
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/pymgrid/envs/base/base.py b/src/pymgrid/envs/base/base.py
@@ -184,10 +184,12 @@ def from_microgrid(cls, microgrid):
 
         """
         try:
-            return cls(microgrid.module_tuples(), add_unbalanced_module=False)
+            modules = microgrid.modules
         except AttributeError:
             assert isinstance(microgrid, NonModularMicrogrid)
             return cls.from_nonmodular(microgrid)
+        else:
+            return cls(modules.to_tuples(), add_unbalanced_module=False)
 
     @classmethod
     def from_nonmodular(cls, nonmodular):