fix: General controller as return, rather than PID

iwishiwasaneagle · Feb 28, 2023 · c7a0f3c · c7a0f3c
1 parent 8d3d78a
commit c7a0f3c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/jdrones/envs/base/basecontrolledenv.py b/src/jdrones/envs/base/basecontrolledenv.py
@@ -7,6 +7,7 @@
 
 import gymnasium
 from gymnasium.core import ActType
+from jdrones.controllers import Controller
 from jdrones.controllers import PID
 from jdrones.envs.base.basedronenev import BaseDroneEnv
 from jdrones.types import State
@@ -56,5 +57,5 @@ def reset(
 
     @staticmethod
     @abc.abstractmethod
-    def _init_controllers(dt: float) -> dict[str, PID]:
+    def _init_controllers(dt: float) -> dict[str, Controller]:
         pass