implements MotorTankGeometry

RocketPy-Team · Oct 28, 2023 · 4825d54 · 4825d54
1 parent a44fc9a
commit 4825d54
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/models/motor.py b/lib/models/motor.py
@@ -28,10 +28,11 @@ class TankFluids(BaseModel, frozen=True):
 class MotorTankGeometry(BaseModel, TankGeometry):
     tank_geometry: dict = {(0, 5): 1}
 
-    def __init__(self, **data):
-        super().__init__(**data)
-        super(TankGeometry, self).__init__()
-        #super(TankGeometry, self).__setattr__("geometry", self.tank_geometry)
+    def __init__(self):
+        super().__init__()
+        attributes = vars(TankGeometry(geometry_dict=self.tank_geometry))
+        for key, value in attributes.items():
+            object.__setattr__(self, key, value)
 
 class MotorTank(BaseModel, frozen=True):
     #Required parameters