Update comments

Author: ratheron

drone_models/_typing.py

@@ -1,6 +1,8 @@
+"""This file is to be remove later as soon as a proper typing is available by the official array-api."""
+
 from typing import Any, TypeAlias
 
 import numpy.typing as npt
 
-Array: TypeAlias = Any  # To be changed to a Protocol later (see array-api#589)
+Array: TypeAlias = Any  # To be changed to array_api_typing later
 ArrayLike: TypeAlias = Array | npt.ArrayLike

drone_models/core.py

@@ -14,7 +14,7 @@
 if TYPE_CHECKING:
     from types import ModuleType
 
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 F = TypeVar("F", bound=Callable[..., Any])

drone_models/first_principles/model.py

@@ -13,7 +13,7 @@
 from drone_models.utils import rotation, to_xp
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 @supports(rotor_dynamics=True)

drone_models/so_rpy/model.py

@@ -14,7 +14,7 @@
 from drone_models.utils import rotation, to_xp
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 @supports(rotor_dynamics=False)

drone_models/so_rpy_rotor/model.py

@@ -14,7 +14,7 @@
 from drone_models.utils import rotation, to_xp
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 @supports(rotor_dynamics=True)

drone_models/so_rpy_rotor_drag/model.py

@@ -14,7 +14,7 @@
 from drone_models.utils import rotation, to_xp
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 # Additional symbols specific to this model
 roll, pitch, yaw = cs.MX.sym("roll"), cs.MX.sym("pitch"), cs.MX.sym("yaw")

drone_models/transform.py

@@ -12,7 +12,7 @@
 from array_api_compat import array_namespace
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 def motor_force2rotor_vel(motor_forces: Array, rpm2thrust: Array) -> Array:

drone_models/utils/__init__.py

@@ -7,7 +7,7 @@
 if TYPE_CHECKING:
     from types import ModuleType
 
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 
 def to_xp(*args: Any, xp: ModuleType, device: Any) -> tuple[Array, ...] | Array:

tests/unit/test_models.py

@@ -22,7 +22,7 @@
 from drone_models.so_rpy_rotor_drag import dynamics as so_rpy_rotor_drag_dynamics
 
 if TYPE_CHECKING:
-    from drone_models._typing import Array  # To be changed to a Protocol later (see array-api#589)
+    from drone_models._typing import Array  # To be changed to array_api_typing later
 
 # For all tests to pass, we need the same precsion in jax as in np
 jax.config.update("jax_enable_x64", True)