monitors and inputs compatibility

Author: chaoming0625

brainpy/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-__version__ = "2.3.2"
+__version__ = "2.3.3"
 
 
 # fundamental supporting modules

brainpy/_src/dyn/runners.py

@@ -361,6 +361,7 @@ def __init__(
       warnings.warn('`fun_inputs` is deprecated since version 2.3.1. '
                     'Define `fun_inputs` as `inputs` instead.',
                     UserWarning)
+    self._fun_inputs = fun_inputs
     if callable(inputs):
       self._inputs = inputs
     else:
@@ -551,6 +552,8 @@ def _step_func_monitor(self, shared):
     return res
 
   def _step_func_input(self, shared):
+    if self._fun_inputs is not None:
+      self._fun_inputs(shared)
     if callable(self._inputs):
       self._inputs(shared)
     else:

brainpy/_src/math/ndarray.py

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