brainstate 0.0.2.post20240910__py2.py3-none-any.whl → 0.0.2.post20240913__py2.py3-none-any.whl

brainstate/init/_regular_inits.py

@@ -16,7 +16,7 @@
 # -*- coding: utf-8 -*-
 
 
-import brainunit as bu
+import brainunit as u
 
 from brainstate import environ
 from ._base import Initializer, to_size
@@ -40,7 +40,7 @@ class ZeroInit(Initializer):
 
   def __call__(self, shape):
     shape = to_size(shape)
-    return bu.math.zeros(shape, dtype=self.dtype)
+    return u.math.zeros(shape, dtype=self.dtype)
 
   def __repr__(self):
     return f"{self.__class__.__name__}(dtype={self.dtype})"
@@ -60,11 +60,11 @@ class Constant(Initializer):
   def __init__(self, value=1., dtype=None):
     super(Constant, self).__init__()
     self.dtype = dtype or environ.dftype()
-    self.value = bu.math.asarray(value, dtype=self.dtype)
+    self.value = value
 
   def __call__(self, shape):
     shape = to_size(shape)
-    return bu.math.full(shape, self.value, dtype=self.dtype)
+    return u.math.full(shape, self.value, dtype=self.dtype)
 
   def __repr__(self):
     return f'{self.__class__.__name__}(value={self.value}, dtype={self.dtype})'
@@ -95,15 +95,15 @@ class Identity(Initializer):
   def __init__(self, value=1., dtype=None):
     super(Identity, self).__init__()
     self.dtype = dtype or environ.dftype()
-    self.value = bu.math.asarray(value, dtype=self.dtype)
+    self.value = value
 
   def __call__(self, shape):
     shape = to_size(shape)
     if isinstance(shape, (tuple, list)):
       if len(shape) > 2:
         raise ValueError(f'Only support initialize 2D weights for {self.__class__.__name__}.')
-    r = bu.math.eye(shape, dtype=self.dtype)
-    r = bu.math.fill_diagonal(r, self.value)
+    r = u.math.eye(shape, dtype=self.dtype)
+    r = u.math.fill_diagonal(r, self.value)
     return r
 
   def __repr__(self):

{brainstate-0.0.2.post20240910.dist-info → brainstate-0.0.2.post20240913.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.0.2.post20240910
+Version: 0.0.2.post20240913
 Summary: A State-based Transformation System for Brain Dynamics Programming.
 Home-page: https://github.com/brainpy/brainstate
 Author: BDP

{brainstate-0.0.2.post20240910.dist-info → brainstate-0.0.2.post20240913.dist-info}/RECORD

@@ -22,7 +22,7 @@ brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1
 brainstate/init/_base.py,sha256=jRTmfoUsH_315vW9YMZzyIn2KDAAsv56SplBnvOyBW0,1148
 brainstate/init/_generic.py,sha256=LB7IQfswOG6X-q0QX5N8T5vZmxdygetsSBQ6iXlZ0oU,7324
 brainstate/init/_random_inits.py,sha256=vNUVDdUOCXTx2i3i1enzxgg1USCzugYd56r0-2lBL-0,15919
-brainstate/init/_regular_inits.py,sha256=u77aSM0BkK9VULFJQZ1lIEYA_sJJzEZBTEttBSJ79RI,3090
+brainstate/init/_regular_inits.py,sha256=2jdIvII_L0FzyRX_Xv81xHjreTt69rXqJ1QPVpMm_Ww,3015
 brainstate/nn/__init__.py,sha256=YJHoI8cXKVRS8f2vUl3Zegp5wm0svMz3qo9JmQJiMQk,2162
 brainstate/nn/_base.py,sha256=lzbZpku3Q2arH6ZaAwjs6bhbV0RcFChxo2UcpnX5t84,8481
 brainstate/nn/_connections.py,sha256=GSOW2IbpJRHdPyF4nFJ2RPgO8y6SVHT1Gn-pbri9pMk,22970
@@ -63,8 +63,8 @@ brainstate/transform/_make_jaxpr_test.py,sha256=K3vRUBroDTCCx0lnmhgHtgrlWvWglJO2
 brainstate/transform/_mapping.py,sha256=G9XUsD1xKLCprwwE0wv0gSXS0NYZ-ZIsv-PKKRlOoTA,3821
 brainstate/transform/_progress_bar.py,sha256=VGoRZPRBmB8ELNwLc6c7S8QhUUTvn0FY46IbBm9cuYM,3502
 brainstate/transform/_unvmap.py,sha256=8Se_23QrwDdcJpFcUnnMgD6EP-4XylbhP9K5TDhW358,3311
-brainstate-0.0.2.post20240910.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.0.2.post20240910.dist-info/METADATA,sha256=gAdKRqW3BdiBR-xvAOdqwKiex6eXywKX0bieOW2_ZZQ,3311
-brainstate-0.0.2.post20240910.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.0.2.post20240910.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.0.2.post20240910.dist-info/RECORD,,
+brainstate-0.0.2.post20240913.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.0.2.post20240913.dist-info/METADATA,sha256=DRuco413fBXd012b5O_sUgatBsggJA_i2aPdBSpn0PU,3311
+brainstate-0.0.2.post20240913.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.0.2.post20240913.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
+brainstate-0.0.2.post20240913.dist-info/RECORD,,