fix bug

Author: chaoming0625

brainpy/_src/initialize/tests/test_regular_inits.py

@@ -20,7 +20,6 @@ def test_one_init(self):
           init = bp.init.OneInit(value=value)
           weights = init(size)
           assert weights.shape == size
-          assert isinstance(weights, bp.math.ndarray)
           assert (weights == value).all()
 
 

brainpy/_src/math/ndarray.py

@@ -93,7 +93,7 @@ def _check_input_array(array):
 
 
 def _return(a):
-  if _return_bp_array and isinstance(a, jax.Array) and a.ndim > 1:
+  if _return_bp_array and isinstance(a, jax.Array) and a.ndim > 0:
     return Array(a)
   return a