fix: numpy integers with linspace (#107)

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · May 13, 2024 · 048cdd0 · 048cdd0
1 parent 110b256
commit 048cdd0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/unxt/_quantity/register_dispatches.py b/src/unxt/_quantity/register_dispatches.py
@@ -5,6 +5,7 @@
 import jax
 import jax.core
 import jax.experimental.array_api as jax_xp
+import numpy as np
 from jax import Device
 
 from quaxed._types import DType
@@ -109,7 +110,7 @@ def full_like(
 def linspace(
     start: Quantity,
     stop: Quantity,
-    num: int,
+    num: int | np.integer,
     *,
     dtype: DType | None = None,
     device: Device | None = None,