update typing

chaobrain · Jun 12, 2024 · 7ee0d89 · 7ee0d89
1 parent 16d4a27
commit 7ee0d89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainunit/math/_compat_numpy_array_creation.py b/brainunit/math/_compat_numpy_array_creation.py
@@ -454,7 +454,7 @@ def zeros_like(a: Union[Quantity, jax.typing.ArrayLike],
 
 @set_module_as('brainunit.math')
 def asarray(
-    a: Union[Quantity, jax.typing.ArrayLike, Sequence[Quantity]],
+    a: Union[Quantity, jax.typing.ArrayLike, Sequence[Quantity], Sequence[jax.typing.ArrayLike]],
     dtype: Optional[jax.typing.DTypeLike] = None,
     order: Optional[str] = None,
     unit: Optional[Unit] = None,