Fix bugs in Python 3.9

chaobrain · Jun 7, 2024 · bc5ba2a · bc5ba2a
1 parent f9dcddf
commit bc5ba2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainunit/math/_compat_numpy.py b/brainunit/math/_compat_numpy.py
@@ -1168,7 +1168,7 @@ def intersect1d(
     ar2: Union[jax.Array, np.ndarray],
     assume_unique: bool = False,
     return_indices: bool = False
-) -> Union[jax.Array, Quantity, tuple[jax.Array | Quantity, jax.Array, jax.Array]]:
+) -> Union[jax.Array, Quantity, tuple[Union[jax.Array, Quantity], jax.Array, jax.Array]]:
   fail_for_dimension_mismatch(ar1, ar2, 'intersect1d')
   unit = None
   if isinstance(ar1, Quantity):