ENH: Reformatted files to pass Ruff linting checks

RocketPy-Team · MateusStano · Mar 23, 2025 · Mar 14, 2025 · Mar 15, 2025 · Mar 15, 2025
commit 85e82e639931e45b6bac1d35484a93e64368d6dd
@@ -2264,7 +2264,9 @@ def __mul__(self, other):  # pylint: disable=too-many-statements
             ):
                 if not self_source_is_array:
                     return Function(lambda x: (self.get_value_opt(x) * other), inputs)
-                source = np.column_stack((self.x_array, np.multiply(self.y_array, other)))
+                source = np.column_stack(
+                    (self.x_array, np.multiply(self.y_array, other))
+                )
                 outputs = f"({self.__outputs__[0]}*{other})"
                 return Function(
                     source,
@@ -2284,15 +2286,21 @@ def __mul__(self, other):  # pylint: disable=too-many-statements
                 and np.array_equal(self.x_array, other.x_array)
                 and np.array_equal(self.y_array, other.y_array)
            ):
-                source = np.column_stack((self.x_array, self.y_array, self.z_array * other.z_array))
+                source = np.column_stack(
+                    (self.x_array, self.y_array, self.z_array * other.z_array)
+                )
                 outputs = f"({self.__outputs__[0]}*{other.__outputs__[0]})"
                 return Function(source, inputs, outputs, interp, extrap)
             elif isinstance(other, NUMERICAL_TYPES) or self.__is_single_element_array(
                 other
             ):
                 if not self_source_is_array:
-                    return Function(lambda x,y: (self.get_value_opt(x,y) * other), inputs)
-                source = np.column_stack((self.x_array, self.y_array, np.multiply(self.z_array, other)))
+                    return Function(
+                        lambda x, y: (self.get_value_opt(x, y) * other), inputs
+                    )
+                source = np.column_stack(
+                    (self.x_array, self.y_array, np.multiply(self.z_array, other))
+                )
                 outputs = f"({self.__outputs__[0]}*{other})"
                 return Function(
                     source,
@@ -2302,7 +2310,9 @@ def __mul__(self, other):  # pylint: disable=too-many-statements
                     extrap,
                 )
             elif callable(other):
-                return Function(lambda x,y: (self.get_value_opt(x,y) * other(x)), inputs)
+                return Function(
+                    lambda x, y: (self.get_value_opt(x, y) * other(x)), inputs
+                )
             else:
                 raise TypeError("Unsupported type for multiplication")
 

@@ -448,7 +448,7 @@ class StochasticAirBrakes(StochasticModel):
     object : AirBrakes
         AirBrakes object to be used for validation.
     drag_coefficient_curve : list, str
-        The drag coefficient curve of the air brakes can account for 
+        The drag coefficient curve of the air brakes can account for
         either the air brakes' drag alone or the combined drag of both
         the rocket and the air brakes.
     drag_coefficient_curve_factor : tuple, list, int, float
@@ -479,7 +479,7 @@ def __init__(
         reference_area=None,
         clamp=None,
         override_rocket_drag=None,
-        deployment_level=(0,0),
+        deployment_level=(0, 0),
     ):
         """Initializes the Stochastic AirBrakes class.
 
@@ -492,7 +492,7 @@ def __init__(
         air_brakes : AirBrakes
             AirBrakes object to be used for validation.
         drag_coefficient_curve : list, str, optional
-            The drag coefficient curve of the air brakes can account for 
+            The drag coefficient curve of the air brakes can account for
             either the air brakes' drag alone or the combined drag of both
             the rocket and the air brakes.
         drag_coefficient_curve_factor : tuple, list, int, float, optional
@@ -534,11 +534,11 @@ def create_object(self):
         """
         generated_dict = next(self.dict_generator())
         air_brakes = AirBrakes(
-            drag_coefficient_curve=generated_dict['drag_coefficient_curve'],
-            reference_area=generated_dict['reference_area'],
-            clamp=generated_dict['clamp'],
-            override_rocket_drag=generated_dict['override_rocket_drag'],
-            deployment_level=generated_dict['deployment_level'],
+            drag_coefficient_curve=generated_dict["drag_coefficient_curve"],
+            reference_area=generated_dict["reference_area"],
+            clamp=generated_dict["clamp"],
+            override_rocket_drag=generated_dict["override_rocket_drag"],
+            deployment_level=generated_dict["deployment_level"],
         )
-        air_brakes.drag_coefficient *= generated_dict['drag_coefficient_curve_factor']
+        air_brakes.drag_coefficient *= generated_dict["drag_coefficient_curve_factor"]
         return air_brakes
@@ -393,11 +393,7 @@ def set_rail_buttons(
             rail_buttons, self._validate_position(rail_buttons, lower_button_position)
         )
 
-    def add_air_brakes(
-        self,
-        air_brakes,
-        controller
-    ):
+    def add_air_brakes(self, air_brakes, controller):
         """Adds an air brake to the stochastic rocket.
 
         Parameters