Added missing classes to init

Signed-off-by: Vlad Iftime <[email protected]>
flexiblepower · Jan 21, 2025 · 9f4bdb0 · 9f4bdb0
1 parent ba68d80
commit 9f4bdb0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/s2python/pebc/__init__.py b/src/s2python/pebc/__init__.py
@@ -3,3 +3,7 @@
 from s2python.pebc.pebc_power_envelope import PEBCPowerEnvelope
 from s2python.pebc.pebc_power_envelope_element import PEBCPowerEnvelopeElement
 from s2python.pebc.pebc_energy_constraint import PEBCEnergyConstraint
+from s2python.generated.gen_s2 import (
+    PEBCPowerEnvelopeConsequenceType,
+    PEBCPowerEnvelopeLimitType,
+)