diff --git a/rocket_twin/drivers/fueling_rocket.py b/rocket_twin/drivers/fueling_rocket.py index 71b0de0..4477684 100644 --- a/rocket_twin/drivers/fueling_rocket.py +++ b/rocket_twin/drivers/fueling_rocket.py @@ -7,7 +7,7 @@ class FuelingRocket(Driver): """Driver that simulates the fueling of a rocket. - + Inputs ------ name: string, @@ -22,6 +22,7 @@ class FuelingRocket(Driver): Outputs ------ """ + def __init__(self, name: str, w_out, dt, owner: Optional["System"] = None, **kwargs): super().__init__(name, owner, **kwargs) diff --git a/rocket_twin/systems/station/__init__.py b/rocket_twin/systems/station/__init__.py index 5e49db6..b485328 100644 --- a/rocket_twin/systems/station/__init__.py +++ b/rocket_twin/systems/station/__init__.py @@ -1,3 +1,3 @@ from rocket_twin.systems.station.station import Station -__all__ = ["Station"] \ No newline at end of file +__all__ = ["Station"] diff --git a/rocket_twin/systems/station/station.py b/rocket_twin/systems/station/station.py index 05d2857..0f8deeb 100644 --- a/rocket_twin/systems/station/station.py +++ b/rocket_twin/systems/station/station.py @@ -12,6 +12,7 @@ class Station(System): Outputs ------ """ + def setup(self): self.add_child(Rocket("rocket")) self.add_child(Pipe("pipe")) @@ -28,4 +29,4 @@ def setup(self): # Design methods dm = self.add_design_method("start") dm.add_unknown("g_tank.weight_p") - dm.add_equation("g_tank.weight_p == g_tank.weight_max") \ No newline at end of file + dm.add_equation("g_tank.weight_p == g_tank.weight_max")