fix bug

chaobrain · Nov 6, 2024 · 2ccd16d · 2ccd16d
1 parent 2404a6f
commit 2ccd16d
Show file tree

Hide file tree

Showing 2 changed files with 869 additions and 210 deletions.
diff --git a/brainunit/math/_fun_array_creation.py b/brainunit/math/_fun_array_creation.py
@@ -604,7 +604,7 @@ def asarray(
   leaf_unit = leaves[0].unit
 
   # get unit
-  if unit is not None:
+  if unit is not None and not leaf_unit.is_unitless:
     assert isinstance(unit, Unit), f'unit must be an instance of Unit, got {type(unit)}'
     leaves = [leaf.in_unit(unit) for leaf in leaves]
   else: