updates

Author: chaoming0625

brainpy/__init__.py

@@ -4,7 +4,7 @@
 
 
 # fundamental modules
-from . import errors, tools, check, modes
+from . import check, errors, tools, modes
 
 # "base" module
 from . import base

brainpy/check.py

@@ -11,24 +11,7 @@
 
 _check = True
 
-_BRAINPYLIB_MINIMAL_VERSION = '0.1.2'
-
-
-def is_checking():
-  """Whether the checking is turn on."""
-  return _check
-
-
-def turn_on():
-  """Turn on the checking."""
-  global _check
-  _check = True
-
-
-def turn_off():
-  """Turn off the checking."""
-  global _check
-  _check = False
+_BRAINPYLIB_MINIMAL_VERSION = '0.1.3'
 
 
 try:
@@ -62,6 +45,7 @@ def turn_off():
 
     ''') from None
 
+
 try:
   import brainpylib
 
@@ -79,3 +63,23 @@ def turn_off():
     f'Please install "brainpylib>={_BRAINPYLIB_MINIMAL_VERSION}" through:\n\n'
     f'>>> pip install brainpylib'
   )
+
+
+def is_checking():
+  """Whether the checking is turn on."""
+  return _check
+
+
+def turn_on():
+  """Turn on the checking."""
+  global _check
+  _check = True
+
+
+def turn_off():
+  """Turn off the checking."""
+  global _check
+  _check = False
+
+
+

brainpy/math/ndarray.py

@@ -94,6 +94,8 @@ def _check_input_array(array):
 class Array(object):
   """Multiple-dimensional array in JAX backend.
   """
+
+  is_brainpy_array = True
   __slots__ = ("_value", "_transform_context")
 
   def __init__(self, value, dtype=None):