lint fixes

Author: Malmahrouqi3

rocketpy/rocket/rocket.py

@@ -1952,38 +1952,42 @@ def add_thrust_eccentricity(self, x, y):
         self.thrust_eccentricity_y = y
         return self
 
-    def add_discrete_controller(self,
-                                controller_function,
-                                refresh_rate,
-                                interactive_objects=None,
-                                initial_observed_variables=None,
-                                name=None
-                                ):
+    def add_discrete_controller(
+        self,
+        controller_function,
+        refresh_rate,
+        interactive_objects=None,
+        initial_observed_variables=None,
+        name=None,
+    ):
 
         controller = _Controller(
-        controller_function=controller_function,
-        sampling_rate=refresh_rate,
-        interactive_objects=interactive_objects,
-        initial_observed_variables=initial_observed_variables,
-        name=name)
+            controller_function=controller_function,
+            sampling_rate=refresh_rate,
+            interactive_objects=interactive_objects,
+            initial_observed_variables=initial_observed_variables,
+            name=name,
+        )
 
         self._add_controllers(controller)
 
         return None
 
-    def add_continuous_controller(self,
-                                controller_function,
-                                interactive_objects=None,
-                                initial_observed_variables=None,
-                                name=None
-                                ):
+    def add_continuous_controller(
+        self,
+        controller_function,
+        interactive_objects=None,
+        initial_observed_variables=None,
+        name=None,
+    ):
 
         controller = _Controller(
             controller_function=controller_function,
             sampling_rate=math.inf,
             interactive_objects=interactive_objects,
             initial_observed_variables=initial_observed_variables,
-            name=name)
+            name=name,
+        )
 
         self._add_controllers(controller)
         return controller

rocketpy/simulation/flight.py

@@ -1592,8 +1592,12 @@ def __init_equations_of_motion(self):
     def __init_controllers(self):
         """Initialize controllers and sensors"""
         self._controllers = self.rocket._controllers[:]
-        self._continuous_controllers = [c for c in self._controllers if math.isinf(c.sampling_rate)]
-        self._discrete_controllers = [c for c in self._controllers if not math.isinf(c.sampling_rate)]
+        self._continuous_controllers = [
+            c for c in self._controllers if math.isinf(c.sampling_rate)
+        ]
+        self._discrete_controllers = [
+            c for c in self._controllers if not math.isinf(c.sampling_rate)
+        ]
         self.sensors = self.rocket.sensors.get_components()
 
         # reset controllable object to initial state (only airbrakes for now)