format codes

Author: chaoming0625

.gitignore

@@ -17,6 +17,7 @@ BrainModels/
 book/
 docs/examples
 docs/apis/jaxsetting.rst
+docs/quickstart/data
 examples/recurrent_neural_network/neurogym
 develop/iconip_paper
 develop/benchmark/COBA/results

brainpy/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 
 
 try:
@@ -15,7 +15,7 @@
 
 
 # fundamental modules
-from . import errors, tools
+from . import errors, tools, check
 
 
 # "base" module

brainpy/math/autograd.py

@@ -1,8 +1,7 @@
 # -*- coding: utf-8 -*-
 
-from typing import Union, Callable, Dict, Sequence
-
 from functools import partial
+from typing import Union, Callable, Dict, Sequence
 
 import jax
 import numpy as np
@@ -41,7 +40,7 @@ def call_func(*args, **kwargs):
     except UnexpectedTracerError as e:
       for v, d in zip(grad_vars, old_grad_vs): v.value = d
       for v, d in zip(dyn_vars, old_dyn_vs): v.value = d
-      raise errors.JaxTracerError(variables=dyn_vars+grad_vars) from e
+      raise errors.JaxTracerError(variables=dyn_vars + grad_vars) from e
     for v, d in zip(grad_vars, new_grad_vs): v.value = d
     for v, d in zip(dyn_vars, new_dyn_vs): v.value = d