Quantity initialization does not call jax.numpy.asarray when mantis…

…sa is a python number
chaobrain · Jan 5, 2025 · 16d4887 · 16d4887
1 parent 1753e86
commit 16d4887
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -227,3 +227,4 @@ cython_debug/
 /docs/apis/changelog.md
 /dist-hist/
 /dist-hist/
+/examples/
diff --git a/brainunit/_base.py b/brainunit/_base.py
@@ -2187,7 +2187,8 @@ def __init__(
                 # skip 'asarray' if dtype is not provided
 
             elif isinstance(mantissa, (jnp.number, numbers.Number)):
-                mantissa = jnp.array(mantissa, dtype=dtype)
+                # mantissa = jnp.array(mantissa, dtype=dtype)
+                mantissa = mantissa
 
             else:
                 mantissa = mantissa