fix: PDS on eom

tmcclintock · Nov 26, 2022 · 1dfb476 · 1dfb476
1 parent 0e53601
commit 1dfb476
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/frispy/equations_of_motion.py b/frispy/equations_of_motion.py
@@ -162,9 +162,7 @@ def compute_forces(
         velocity: np.ndarray,
         ang_velocity: np.ndarray,
     ) -> Dict[str, Union[float, np.ndarray, Dict[str, np.ndarray]]]:
-        """
-        Compute the lift, drag, and gravitational forces on the disc.
-        """
+        """Computes the lift, drag, and gravitational forces on the disc."""
         res = self.geometric_quantities(phi, theta, velocity, ang_velocity)
         aoa = res["angle_of_attack"]
         vhat = velocity / np.linalg.norm(velocity)
@@ -187,9 +185,7 @@ def compute_torques(
         velocity: np.ndarray,
         res: Dict[str, Union[float, np.ndarray, Dict[str, np.ndarray]]],
     ) -> Dict[str, Union[float, np.ndarray, Dict[str, np.ndarray]]]:
-        """
-        Compute the torque around each principle axis.
-        """
+        """Computes the torque around each principle axis."""
         aoa = res["angle_of_attack"]
         res["torque_amplitude"] = self.torque_per_v2 * (velocity @ velocity)
         wx, wy, wz = res["w"]