[dyn] add reduce models, HH-type models and channels (#408)

[dyn] add reduce models, HH-type models and channels

brainpy/_src/__init__.py

@@ -1 +0,0 @@
-# -*- coding: utf-8 -*-

brainpy/_src/dyn/_docs.py

@@ -18,6 +18,13 @@
     has_ref_var: bool. Whether has the refractory variable. Default is ``False``.
 '''.strip()
 
+if_doc = '''
+    V_rest: float, ArrayType, callable. Resting membrane potential.
+    R: float, ArrayType, callable. Membrane resistance.
+    tau: float, ArrayType, callable. Membrane time constant.
+    V_initializer: ArrayType, callable. The initializer of membrane potential.
+'''.strip()
+
 lif_doc = '''
     V_rest: float, ArrayType, callable. Resting membrane potential.
     V_reset: float, ArrayType, callable. Reset potential after spike.

brainpy/_src/dyn/base.py

@@ -28,6 +28,7 @@ def __init__(
       keep_size: bool = False,
       mode: bm.Mode = None,
       name: str = None,
+      method: str = 'exp_auto'
   ):
     super().__init__(size=size,
                      mode=mode,
@@ -37,6 +38,9 @@ def __init__(
     # axis names for parallelization
     self.sharding = sharding
 
+    # integration method
+    self.method = method
+
     # the before- / after-updates used for computing
     self.before_updates: Dict[str, Callable] = bm.node_dict()
     self.after_updates: Dict[str, Callable] = bm.node_dict()
@@ -109,21 +113,21 @@ def __init__(
       keep_size: bool = False,
       mode: Optional[bm.Mode] = None,
       name: Optional[str] = None,
+      method: str = 'exp_auto',
 
       spk_fun: Callable = bm.surrogate.InvSquareGrad(),
       spk_type: Any = None,
       detach_spk: bool = False,
-      method: str = 'exp_auto',
   ):
     super().__init__(size=size,
                      mode=mode,
                      keep_size=keep_size,
                      name=name,
-                     sharding=sharding)
+                     sharding=sharding,
+                     method=method)
 
     self.spk_fun = is_callable(spk_fun)
     self.detach_spk = detach_spk
-    self.method = method
     self._spk_type = spk_type
 
   @property

brainpy/_src/dyn/channels/__init__.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+
+"""
+
+Access through ``brainpy.channels``.
+"""
+
+from . import base, Ca, IH, K, Na, KCa, leaky
+
+__all__ = []
+__all__ += base.__all__
+__all__ += K.__all__
+__all__ += Na.__all__
+__all__ += Ca.__all__
+__all__ += IH.__all__
+__all__ += KCa.__all__
+__all__ += leaky.__all__
+
+from .base import *
+from .K import *
+from .Na import *
+from .IH import *
+from .Ca import *
+from .KCa import *
+from .leaky import *

brainpy/_src/channels/tests/test_Ca.py → brainpy/_src/dyn/channels/tests/test_Ca.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import Ca
+from brainpy._src.dyn.channels import Ca
 
 class Test_Ca(parameterized.TestCase):
   def test_Ca(self):

brainpy/_src/channels/tests/test_IH.py → brainpy/_src/dyn/channels/tests/test_IH.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import IH, Ca
+from brainpy._src.dyn.channels import IH, Ca
 
 
 class Test_IH(parameterized.TestCase):

brainpy/_src/channels/tests/test_K.py → brainpy/_src/dyn/channels/tests/test_K.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import K
+from brainpy._src.dyn.channels import K
 
 class Test_K(parameterized.TestCase):
   bm.random.seed(1234)

brainpy/_src/channels/tests/test_KCa.py → brainpy/_src/dyn/channels/tests/test_KCa.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import KCa, Ca
+from brainpy._src.dyn.channels import KCa, Ca
 
 class Test_KCa(parameterized.TestCase):
   bm.random.seed(1234)

brainpy/_src/channels/tests/test_Na.py → brainpy/_src/dyn/channels/tests/test_Na.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import Na
+from brainpy._src.dyn.channels import Na
 
 
 class Test_Na(parameterized.TestCase):

brainpy/_src/channels/tests/test_leaky.py → brainpy/_src/dyn/channels/tests/test_leaky.py

@@ -4,7 +4,7 @@
 import brainpy as bp
 import brainpy.math as bm
 from absl.testing import parameterized
-from brainpy._src.channels import leaky
+from brainpy._src.dyn.channels import leaky
 
 class Test_Leaky(parameterized.TestCase):
   bm.random.seed(1234)