ENH: add missing type hinting

AndreaBlengino · Oct 4, 2024 · f3f2c0a · f3f2c0a
1 parent bbbc383
commit f3f2c0a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gearpy/solver.py b/gearpy/solver.py
@@ -247,17 +247,17 @@ def _compute_angular_position_and_speed(self):
             self._transmit_angular_position(gear_ratio=gear_ratio, i=i)
             self._transmit_angular_speed(gear_ratio=gear_ratio, i=i)
 
-    def _transmit_angular_position(self, gear_ratio, i):
+    def _transmit_angular_position(self, gear_ratio: float, i: int):
 
         self.__powertrain.elements[i].angular_position = \
             gear_ratio*self.__powertrain.elements[i + 1].angular_position
 
-    def _transmit_angular_speed(self, gear_ratio, i):
+    def _transmit_angular_speed(self, gear_ratio: float, i: int):
 
         self.__powertrain.elements[i].angular_speed = \
             gear_ratio*self.__powertrain.elements[i + 1].angular_speed
 
-    def _transmit_angular_acceleration(self, gear_ratio, i):
+    def _transmit_angular_acceleration(self, gear_ratio: float, i: int):
 
         self.__powertrain.elements[i].angular_acceleration = \
             gear_ratio * \