brainstate 0.1.1__py2.py3-none-any.whl → 0.1.2__py2.py3-none-any.whl

brainstate/random/_rand_seed_test.py

@@ -13,14 +13,12 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 import unittest
 
 import jax.numpy as jnp
 import jax.random
 
-import brainstate as bst
+import brainstate
 
 
 class TestRandom(unittest.TestCase):
@@ -28,23 +26,23 @@ class TestRandom(unittest.TestCase):
     def test_seed2(self):
         test_seed = 299
         key = jax.random.PRNGKey(test_seed)
-        bst.random.seed(key)
+        brainstate.random.seed(key)
 
         @jax.jit
         def jit_seed(key):
-            bst.random.seed(key)
-            with bst.random.seed_context(key):
-                print(bst.random.DEFAULT.value)
+            brainstate.random.seed(key)
+            with brainstate.random.seed_context(key):
+                print(brainstate.random.DEFAULT.value)
 
         jit_seed(key)
         jit_seed(1)
         jit_seed(None)
-        bst.random.seed(1)
+        brainstate.random.seed(1)
 
     def test_seed(self):
         test_seed = 299
-        bst.random.seed(test_seed)
-        a = bst.random.rand(3)
-        bst.random.seed(test_seed)
-        b = bst.random.rand(3)
+        brainstate.random.seed(test_seed)
+        a = brainstate.random.rand(3)
+        brainstate.random.seed(test_seed)
+        b = brainstate.random.rand(3)
         self.assertTrue(jnp.array_equal(a, b))

{brainstate-0.1.1.dist-info → brainstate-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.1.1
+Version: 0.1.2
 Summary: A ``State``-based Transformation System for Program Compilation and Augmentation.
 Home-page: https://github.com/chaobrain/brainstate
 Author: BrainState Developers

{brainstate-0.1.1.dist-info → brainstate-0.1.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
-brainstate/__init__.py,sha256=RrDTTcA9N5s3GU2fxXMk4GwGmMuHdkhf4i2Ks98INFY,1496
-brainstate/_compatible_import.py,sha256=MSLUh2xFmboTpwqVnBy5X8dOlNi2fafwFnaU_JJIXQw,4931
-brainstate/_state.py,sha256=bHV6gEFQbYMPiQNcdTxGTMGFaHuQrY9wIR3ftCyeFJk,60749
+brainstate/__init__.py,sha256=50znZ0wsJBdgX9Wvlv4xvsuNGp4XdgR60LTFrplmBz0,1496
+brainstate/_compatible_import.py,sha256=96CR40jbcrvaBXO0Hrlm7Cpde-4nRj-6z-P753bitFc,5001
+brainstate/_state.py,sha256=o5Kk4bGwVz6Dfj9dlmZqdh6zUXcz6Tvc6WOjH9ajlIU,60716
 brainstate/_state_test.py,sha256=b6uvZdVRyC4n6-fYzmHNry1b-gJ6zE_kRSxGinqiHaw,1638
 brainstate/_utils.py,sha256=j-b239RHfC5BnvhGbSExQpdY21LrMEyWMSHBdNGThOI,1657
 brainstate/environ.py,sha256=VgtG0S_aR1g_1gplRWg_v2ZrcS-F6LZk35BPCBgsIIA,17660
@@ -12,107 +12,107 @@ brainstate/transform.py,sha256=OvshYpPnp3YXPG6riy15Ve7jcX8t0aaATl1xZnsFeic,858
 brainstate/typing.py,sha256=XEOzcvMos8jcVeOf2rZb20M4A89xUEUpEIHu9qit2L8,10485
 brainstate/augment/__init__.py,sha256=Q9-JIwQ1FNn8VLS1MA9MrSylbvUjWSw98whrI3NIuKo,1229
 brainstate/augment/_autograd.py,sha256=zMMLZOidQq2p96wzEOgR-MynV2JH1l1AnoJ28eVwD-s,30100
-brainstate/augment/_autograd_test.py,sha256=S2eEgrwTzdSi3u2nKE3u37WSThosLwx1WCP9ptJAGKo,44060
+brainstate/augment/_autograd_test.py,sha256=UhNd41luca_Kj9a8byL3Dq64Ta55WU-je-LCiK_z0Vc,45060
 brainstate/augment/_eval_shape.py,sha256=mYRB5clSRHdXX0c4c5_WDIIBtRl4a9-xl8Tg60mgLBI,3854
-brainstate/augment/_eval_shape_test.py,sha256=LFOJx7CWltmRLXdGY175UebLwtEMz2CzJ_gLqMZsJTw,1393
+brainstate/augment/_eval_shape_test.py,sha256=WXrmZKmnykmYPveRfZbrDF0sFm2_PTr982yl4JM7Ebg,1390
 brainstate/augment/_mapping.py,sha256=AvGt8qhOdBgRJyWZn7J5JnOvxZc7XfZ_JKXaECOQjIw,43500
-brainstate/augment/_mapping_test.py,sha256=OLkd9obwcy5K8jHpiQLT1o5c4rVG7tfg5-CYdMJvJWQ,14711
+brainstate/augment/_mapping_test.py,sha256=An5iSqrAXq_OVTvY3GFTjsB9AvFZ4j6AWNZQ_WWfra4,15319
 brainstate/augment/_random.py,sha256=bkngsIk6Wwi3e46I7YSbBjLCGAz0Q3WuadUH4mqTjbY,5348
 brainstate/compile/__init__.py,sha256=fQtG316MLkeeu1Ssp54Kghw1PwbGK5gNq9yRVJu0wjA,1474
 brainstate/compile/_ad_checkpoint.py,sha256=HM6L90HU0N84S30uJpX8wqTO0ZcDnctqmwf8qFlkDYo,9358
-brainstate/compile/_ad_checkpoint_test.py,sha256=R1I76nG4zIqb6g3M_VxWts7rUC1OHJCjtQhPkcbXodk,1746
+brainstate/compile/_ad_checkpoint_test.py,sha256=XGqC8kjwKd64RPgBWXHAE3t8NdIXVUqmGDfrg5HqoVw,1743
 brainstate/compile/_conditions.py,sha256=fFSJWRFZxriAZFASj2lvlXpxbH323XcfsoB9ni2CYZ0,10183
-brainstate/compile/_conditions_test.py,sha256=s9LF6h9LvigvgxUIugTqvgCHBIU8TXS1Ar1OlIxXfrw,8389
+brainstate/compile/_conditions_test.py,sha256=bh8_D49Fl-s_xFYkO4joZJPgFGhw_vd6Wo8QB1HDKwE,8640
 brainstate/compile/_error_if.py,sha256=l8flahWEui9-9IavtZQjyutIeR8DU0jieahRvAXUxuw,2651
-brainstate/compile/_error_if_test.py,sha256=OdJG483IIdOrCHxtHd49OHfOxCSnSkk7GdAUOzSt8bE,2044
+brainstate/compile/_error_if_test.py,sha256=PjTIQOflegqnqJb9kO5YqwjW1OfrBmAtWlY_xn8rtDY,1897
 brainstate/compile/_jit.py,sha256=TN6Fk7qTHpZ2rzfBf4uyL9oFoU_IEpaDQhn1R_d_bd4,13858
 brainstate/compile/_jit_test.py,sha256=zD7kck9SQJGmUDolh9P4luKwQ21fBGje1Z4STTEXIuA,4135
 brainstate/compile/_loop_collect_return.py,sha256=qOYGoD2TMZd2Px6y241GWNwSjOLyaKlvzOwJWIPAbxg,23341
-brainstate/compile/_loop_collect_return_test.py,sha256=r-q-D2fYuO9oIGKCu7ZUZDlZj-RlPthdn7fmP04TxbQ,1804
+brainstate/compile/_loop_collect_return_test.py,sha256=pEJdcOthEM17q5kXYhgR6JfzzqibijD_O1VPXVe_Ml4,1808
 brainstate/compile/_loop_no_collection.py,sha256=tPkSxee41VexWEALpN1uuT78BDrX3uT1FHv8WZtov4c,7549
-brainstate/compile/_loop_no_collection_test.py,sha256=oStB1CSG_iLp9sHdXd1hJNFvlxbzjck9Iy4sABoJDj4,1419
-brainstate/compile/_make_jaxpr.py,sha256=-864jBQjZCUft_J7m82C9yQSZ4TnTwYliEnOScV9_zM,33120
-brainstate/compile/_make_jaxpr_test.py,sha256=3rfb1oz0u1wHPmnppnmo2BlUBbEE02FlbvD6Rehub9I,4290
+brainstate/compile/_loop_no_collection_test.py,sha256=ivavF59xep_g9bV1SSdXd5E1f6nhc7EUBfcbgHpxbfg,1419
+brainstate/compile/_make_jaxpr.py,sha256=nW5VV9fcqD6bo0GBWxGT2005ANox1ym7ChhGV_q5Aig,33463
+brainstate/compile/_make_jaxpr_test.py,sha256=wvZHl7ecQrmD6C_Nxy8z64l6r3Tj1kQEM54UNIU1Er4,4416
 brainstate/compile/_progress_bar.py,sha256=iTp-CkcK5aMbiA0ZdlSfEXXrc4_anifYhyDzl19dksI,7488
 brainstate/compile/_unvmap.py,sha256=9S42MeTmFJa8nfBI_AjEfrAdUsDmM3KFat59O8zXIEw,4120
 brainstate/compile/_util.py,sha256=QD7lvS4Zb3P2HPNIm6mG8Rl_Lk2tQj0znOCPcH95XnI,6275
 brainstate/functional/__init__.py,sha256=j6-3Er4fgqWpvntzYCZVB3e5hoz-Z3aqvapITCuDri0,1107
 brainstate/functional/_activations.py,sha256=3ym1oDdftVxYlHIXWdS7XONiDFaDklqokIioi4qrsHo,21688
-brainstate/functional/_activations_test.py,sha256=T___RlSrIfXwlkw8dg5A9EZMTZGDzv3a2evUwq_nYFg,13034
+brainstate/functional/_activations_test.py,sha256=66PTMI99pfi1hr2rfFixg2OlUN0WME9628T7U2NF0iI,13475
 brainstate/functional/_normalization.py,sha256=_F5u9iJm8KJfxlnmV_LwJXtrAZOJj357aXc_sfGe4CQ,2567
 brainstate/functional/_others.py,sha256=iAqvLCmXQz6C80EjE0-5bwXCBKHoKqsggneiOAXNmQo,1699
 brainstate/functional/_spikes.py,sha256=YJ4pa5LlywximSE6ZAAs9g2k-8cO95mwxdD29HhT-oM,7051
 brainstate/graph/__init__.py,sha256=noo4TjBg6iEhjjwk0sAGUhR7Ge-z8Vnc2rLYUvnqttw,1295
 brainstate/graph/_graph_node.py,sha256=zxtfzv6RXs9ylTerW5scUZ4mHic66df5WTHCrUzHLDA,6836
-brainstate/graph/_graph_node_test.py,sha256=BFGfdzZFDHI0XK7hHotSVWKt3em1taGvn8FHF9NCXx8,2702
+brainstate/graph/_graph_node_test.py,sha256=g6nrCCI-awfxhZLnjmYhfQvynumWJjz7fPxgIF-VSr4,2832
 brainstate/graph/_graph_operation.py,sha256=gnLZB5YLgsnwVAZugv_5sryzU9_QYyz2r0-D5y1ZG7Q,63405
-brainstate/graph/_graph_operation_test.py,sha256=zjvpKjQAFWtw8YZuqOk_jmlZNb_-E8oPyNx57dyc8jI,18556
+brainstate/graph/_graph_operation_test.py,sha256=nxMwEoaRZpTkIeR3h_Aduuys0Bm1DRlYLd_CXkErTSg,19715
 brainstate/init/__init__.py,sha256=R1dHgub47o-WJM9QkFLc7x_Q7GsyaKKDtrRHTFPpC5g,1097
 brainstate/init/_base.py,sha256=A5VIJV_xPyCD4Lo4RXe2WPlWvADAy_UPhexuwzp-6EI,1633
 brainstate/init/_generic.py,sha256=o60EPJZe7laJgPsgrnorl3Km_Lc_Lr4qQQKmu6g5G4o,7992
 brainstate/init/_random_inits.py,sha256=GYovk2h2Lc1jY5ByW4R8OoZ71jbaE3BYGTfDw4-a7yo,18209
-brainstate/init/_random_inits_test.py,sha256=lBL2RQdBSZ88Zqz4IMdbHJMvDi7ooZq6caCpHfNtIRk,5197
+brainstate/init/_random_inits_test.py,sha256=c1pUdWXQxL-seEvjO_8fBkaDL6T63b6ALNfQrNuuBbg,5287
 brainstate/init/_regular_inits.py,sha256=tcZDTXR8jiRqERuJleg_9JRGGzJj-V3LhTMcr_gt_k0,3149
-brainstate/init/_regular_inits_test.py,sha256=tJl4aOkclllJIfKzJTbc0cfYCw2SoBsx8_G123RnqbU,1842
+brainstate/init/_regular_inits_test.py,sha256=bvv0AOfLOEP0BIQIBLztKw3EPyEp7n2fHW8PUrmWuHk,1820
 brainstate/nn/__init__.py,sha256=dUEH_S5jRTq0nY1xNkpN_4MQrmHXHGSmGp3xgGenqNc,1924
 brainstate/nn/_collective_ops.py,sha256=rWcjqaP0rW6sXdhE0fjtDi4twmq5VvhNEg8AoiQ-tDU,17390
-brainstate/nn/_collective_ops_test.py,sha256=vvyXNPluFa0Ii_sxVUTwsdMwiEITSXvRA1o7-HoPANk,1367
+brainstate/nn/_collective_ops_test.py,sha256=bwq0DApcsk0_2xpxMl0_e2cGKT63g5rSngpigCm07ps,1409
 brainstate/nn/_common.py,sha256=Pt8P1LgE0qW3QnfX8CQQGH3yZ78RL7NkIqqYEw_z8xs,7096
 brainstate/nn/_exp_euler.py,sha256=ndDB43PM4jsZKu_zdLTZ2-ojnuNrg55LZap23oBTtdA,3493
-brainstate/nn/_exp_euler_test.py,sha256=clwRD8QR71k1jn6NrACMDEUcFMh0J9RTosoPnlYWUkw,1242
+brainstate/nn/_exp_euler_test.py,sha256=XD--qMbGHrHa3WtcPMmJKk59giDcEhSqZuBOmTNYUr8,1227
 brainstate/nn/_module.py,sha256=ksbmO_-bZzaP8XTRAcN2oi2ZMdSuSv-XVeWc2VJncVc,12808
-brainstate/nn/_module_test.py,sha256=V4ZhiY_zYPvArkB2eeOTtZcgQrtlRyXKMbS1AJH4vC8,8893
+brainstate/nn/_module_test.py,sha256=_RKHY8LHg6nbFHkvPAsEPVpp9rW6XfrcF8YD68u4o4w,8956
 brainstate/nn/_utils.py,sha256=epfELIy1COgdS9z5be-fmbFhagNugcIHpw4ww-HlkSY,3123
 brainstate/nn/metrics.py,sha256=TXCB_yGQzamklJCI5FGOAZ5dihtY-gjYlSi2SCLC3LA,14700
 brainstate/nn/_dyn_impl/__init__.py,sha256=J1hMqNey1921tZppCjuHApB6oc-5aGhWXyOXh96CeWM,1322
 brainstate/nn/_dyn_impl/_dynamics_neuron.py,sha256=6xsGF2Oy_oD6bcTdfwqnHBtfdAJREIl5qvNUZKZRlnQ,27482
-brainstate/nn/_dyn_impl/_dynamics_neuron_test.py,sha256=Tfzrzu7udGrLJGnqItiLWe5WT0dgduvYOgzGCnaPJQg,6317
+brainstate/nn/_dyn_impl/_dynamics_neuron_test.py,sha256=QF8pixUqA5Oj7MrNi2NR8VAnfGpAvNpwV2mBc3e_pTY,6393
 brainstate/nn/_dyn_impl/_dynamics_synapse.py,sha256=b2QMEzasbXTp7k1XWNg7nRt4-WddCym0CSEErjPRIWA,27449
-brainstate/nn/_dyn_impl/_dynamics_synapse_test.py,sha256=t5i-HV0ii9sUNzWTEv04o26QVtQ-mCdMJcFq2MD755A,4981
+brainstate/nn/_dyn_impl/_dynamics_synapse_test.py,sha256=xmCWFxZUIM2YtmW5otKnADGCCK__4JpXmSYcZ3wzlQM,4994
 brainstate/nn/_dyn_impl/_inputs.py,sha256=Fe3tgSwJMrq2HfZJq9CTgNy24Ji6tnLCqB4hkhaY_K0,20675
 brainstate/nn/_dyn_impl/_rate_rnns.py,sha256=SXlrHZhih4LD99Jg8OwIWur5UQbd6E_zotkHvkFFstc,20715
-brainstate/nn/_dyn_impl/_rate_rnns_test.py,sha256=gNgtr-a4ZiU1XF9wFG1HiJ9fLosfWchVR9Zn1x39xt4,2452
+brainstate/nn/_dyn_impl/_rate_rnns_test.py,sha256=__hhx7e6LX_1mDLLQyIi4TNCaFAWnOVSTIgwHNjzf2g,2444
 brainstate/nn/_dyn_impl/_readout.py,sha256=UzaJ6WIMpNjqHolhjHTpyCZ5E2nSDucvuV5PBW7rOPs,7151
-brainstate/nn/_dyn_impl/_readout_test.py,sha256=R9JJPRvy3mAHSv8n1Hzjk2kBSDjBzJNbS83ystll86s,2109
+brainstate/nn/_dyn_impl/_readout_test.py,sha256=L2T0-SkiACxkY_I5Pbnbmy0Zw3tbpV3l5xVzAw42f2g,2136
 brainstate/nn/_dynamics/__init__.py,sha256=j1HSWu01wf5-KjSaNhBC9utVGDALOhUsFPrLPcPPDsM,1208
 brainstate/nn/_dynamics/_dynamics_base.py,sha256=stBAkb8f973vJe66p0tve-0AO_j008BCpAA96FIPnTI,43363
-brainstate/nn/_dynamics/_dynamics_base_test.py,sha256=gXMwENqqSvyZbMpLP0QtYndJ_h39dF5gIeiiSbMAjTk,2721
+brainstate/nn/_dynamics/_dynamics_base_test.py,sha256=w7AV57LdhbBNYprdFpKq8MFSCbXKVkGgp_NbL3ANX3I,2769
 brainstate/nn/_dynamics/_projection_base.py,sha256=O86VfZkd8Qd5_8CDK-4eCGS6Jkp1CKyW6J9__bQnmo0,12893
 brainstate/nn/_dynamics/_state_delay.py,sha256=kSgel1OSFB92WcfGTsIjgb-KW389ydXTcLbTUjylU-A,16867
 brainstate/nn/_dynamics/_synouts.py,sha256=Py9DM3o-JmkoqYyQM5vl89-SBqTBRdk2CUYgYHYu3No,4453
-brainstate/nn/_dynamics/_synouts_test.py,sha256=V_jDswRN4VvEXD-2yJO3VA1TALgX0HK6oPBQiUntOWc,2266
+brainstate/nn/_dynamics/_synouts_test.py,sha256=sfjotlS--4hT22Vb5RfmpfmXABa8z-VdxZYUSocMmlU,2244
 brainstate/nn/_elementwise/__init__.py,sha256=PK8oq1K_EG2941AiUyLxCWoRdWvMO3yt8ZJbw3Lkhu8,935
 brainstate/nn/_elementwise/_dropout.py,sha256=ymr87YYA5x5C_c9HzNLGPJK_ebJ22Y-KsK-gZJY_2MU,17733
-brainstate/nn/_elementwise/_dropout_test.py,sha256=k6aB5v8RYMoV5w8UV9UNSFhaQTV7woS6jx3SNESuCRs,4383
+brainstate/nn/_elementwise/_dropout_test.py,sha256=9i2ZW5by0S9zfVHmZKLN0WpI1bXH6h9-QLXamG6lVXA,4432
 brainstate/nn/_elementwise/_elementwise.py,sha256=yO6BukJFLfnICPELEFp2kfutFM8Jeg7TaERXFp-aNrk,33304
-brainstate/nn/_elementwise/_elementwise_test.py,sha256=SZI9jB39sZ5SO1dpWGW-PhodthwN0GU9FY1nqf2fWcs,5341
+brainstate/nn/_elementwise/_elementwise_test.py,sha256=_dd9eX2ZJ7p24ahuoapCaRTZ0g1boufXMyqHFx1d4WY,5688
 brainstate/nn/_event/__init__.py,sha256=OLteeA3d1Kq8yi3XETA-rFYtYpojZRpY1J_h5MpH3rY,919
 brainstate/nn/_event/_fixedprob_mv.py,sha256=MpVCiYGiDC8GAuJWtYXKrpzP9yDtnYHolHlwyzoQ1r4,6619
-brainstate/nn/_event/_fixedprob_mv_test.py,sha256=AAtzAY9K3YS-n_5oEOTR5EUnEtWZXGnOw1xc4oOLXW8,3965
+brainstate/nn/_event/_fixedprob_mv_test.py,sha256=qbRBh-MpMtEOsg492gFu2w9-FOP9z_bXapm-Q0gLLYM,3929
 brainstate/nn/_event/_linear_mv.py,sha256=KslIdse7KZ7FWm5qcOMC-COM__1rB-sUJmXvlXY11y8,2683
-brainstate/nn/_event/_linear_mv_test.py,sha256=XMffPFMXwSryAW1gLxIEXE8eC7DJSC3aWRMDuDrLem8,3900
+brainstate/nn/_event/_linear_mv_test.py,sha256=ZCM1Zy6mImQfCfdZOGnTwkiLLPXK5yalv1Ts9sWZuPA,3864
 brainstate/nn/_interaction/__init__.py,sha256=TTY_SeNrdx4VnUSw6vdyl02OHdS9Qs15cWBp6kjsyNQ,1289
 brainstate/nn/_interaction/_conv.py,sha256=QMZR4HRAYe8Nzr3qExp0PWr6wuUp2rh5YxmPq1r1vcY,18461
-brainstate/nn/_interaction/_conv_test.py,sha256=VJacu1afSmxFpgKerjMA2r3kwXAhf7aVhP6QdgJbqyY,8697
+brainstate/nn/_interaction/_conv_test.py,sha256=2lcUTG7twkyhuyKwuBux-NgU8NU_W4Cp1-G8EyDJ_uk,8862
 brainstate/nn/_interaction/_embedding.py,sha256=8HC-J3RhvGGrvqA0-BBaFDk6tmlunPP6bNohjRSQzYo,2151
 brainstate/nn/_interaction/_linear.py,sha256=jk765kASgwVG1r-xjoM49Thfa1Cu32mscy7ci79bjDI,14729
-brainstate/nn/_interaction/_linear_test.py,sha256=QfCR8SBBed9OnSY-AmQ0kJqoggDA3Xem0dRJ0BusxLU,2872
+brainstate/nn/_interaction/_linear_test.py,sha256=eIS-VCR3QmXB_byO1Uexg65Pv48CBRUA_Je-UGrFVTY,2925
 brainstate/nn/_interaction/_normalizations.py,sha256=4yoU04wK21_zk6Egpwb9YN_pSWUH5Qa6hJ1bLxX7wQw,37400
-brainstate/nn/_interaction/_normalizations_test.py,sha256=2p1Jf8nA999VYGWbvOZfKYlKk6UmL0vaEB76xkXxkXw,2438
+brainstate/nn/_interaction/_normalizations_test.py,sha256=JTqpH265GT0eKw3hXmT0qC8ZM5NkUe98of0uYS7_5Us,2456
 brainstate/nn/_interaction/_poolings.py,sha256=2hJDACuHv3CkO-nyyAqVZp7Sa3UFa_ROFGiYmhXWqKk,46982
-brainstate/nn/_interaction/_poolings_test.py,sha256=wmd5PngZ3E9tNyF3s0xk-DoDR5yFqpTi9A6nbNoIqn4,7429
+brainstate/nn/_interaction/_poolings_test.py,sha256=7-jEs_UcMNWm8q6NbrRRX3bTn14jBjGiAX0KBaROng0,7510
 brainstate/optim/__init__.py,sha256=7Ao0LCtDNAoxSRSXiLLKnd1_4mR2GSExizpN38il-Fo,1195
 brainstate/optim/_base.py,sha256=P37k8w46iQZZZnFLa5OF83Sb8DLQfYMez9ZRObBqVsE,1835
 brainstate/optim/_lr_scheduler.py,sha256=0t7kl35MLmc0UcSu587nI28cI3SohYhy_19Al-5dNTM,15289
-brainstate/optim/_lr_scheduler_test.py,sha256=QJSYfvd7llHOGWY6C2Jc_D3QbH-yZoRK8mSn5osmSes,1776
+brainstate/optim/_lr_scheduler_test.py,sha256=gzzd_rzgtDWzF-93mujg043jv2wO9IceXGKiIYw5_ko,1790
 brainstate/optim/_optax_optimizer.py,sha256=RcO5N5JO_U0P4Q82waQuwMS7yWR-oP18eUwgzXJX0EY,5274
-brainstate/optim/_optax_optimizer_test.py,sha256=DAomE8Eu3dn4gh1S3EZ_u4pW4rhcl16vWPbnDcN3Rs4,1762
+brainstate/optim/_optax_optimizer_test.py,sha256=GVcszbNxBZ4z54bS5r-zDzP8gQZl7s5IsEZEd6G2gLo,1775
 brainstate/optim/_sgd_optimizer.py,sha256=Fn4CdGaDfV4qc_4gacCcGSjit5-ZrsBbQP-KB0Ocdac,46110
 brainstate/random/__init__.py,sha256=c5q-RC3grRIjx-HBb2IhKZpi_xzbFmUUxzRAzqfREic,1045
 brainstate/random/_rand_funs.py,sha256=c4xiY2NeMizSslxbWOa-QJZ3h-LDfsgH4fbvBMSNLWg,137646
-brainstate/random/_rand_funs_test.py,sha256=abO5lSoPBgBcg6ecFE1qnCg98__QGa68GSYC5pQW5QI,19438
+brainstate/random/_rand_funs_test.py,sha256=G8BuxDjBSeE-Mh7KuwVk6mPQhHE0m1R5HDFljEOIdzg,20669
 brainstate/random/_rand_seed.py,sha256=1ZdfFZWyOhpd72EDdEDmpkp3yoLVwdv-sGI9BwiZfzI,5949
-brainstate/random/_rand_seed_test.py,sha256=Qibcs-ZqCvj1LuucmQ8H00B_HBNhf2f6un0aUdNZNTw,1518
+brainstate/random/_rand_seed_test.py,sha256=waXXfch57X1XE1zDnCRokT6ziZOK0g-lYE80o6epDYM,1536
 brainstate/random/_rand_state.py,sha256=dTO7wqmuTYRdPy7ItsrK-7aNt5QQXOsZ4XwiH7mzmy8,55170
 brainstate/random/_random_for_unit.py,sha256=kGp4EUX19MXJ9Govoivbg8N0bddqOldKEI2h_TbdONY,2057
 brainstate/util/__init__.py,sha256=zwdw0253tY2kgWHHPPwaU6JHvjw4rioUnRmSAWF4Po0,1569
@@ -126,8 +126,8 @@ brainstate/util/_pretty_table.py,sha256=uJVaamFGQ4nKP8TkEGPWXHpzjMecDo2q1Ah6XtRj
 brainstate/util/_scaling.py,sha256=U6DM-afPrLejiGqo1Nla7z4YbTBVicctsBEweurr_mk,7524
 brainstate/util/_struct.py,sha256=7HbbQNrZ3zxYw93MU1bUZ9ZPBKftYOVKuXEochLSErw,17479
 brainstate/util/filter.py,sha256=skuq51y0P5-N611fQ-zB2QyaJIFqN_WbROzZHv0RgOc,14086
-brainstate-0.1.1.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.1.1.dist-info/METADATA,sha256=_vaI9i2ZItieDpBQMXDI0yodxtyV_Fi4F2pb6riQPR4,4135
-brainstate-0.1.1.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.1.1.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.1.1.dist-info/RECORD,,
+brainstate-0.1.2.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.1.2.dist-info/METADATA,sha256=Mv44muheQeXJ67RF5GQrkSxVWfj4VAZnkse6_7z5OTY,4135
+brainstate-0.1.2.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.1.2.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
+brainstate-0.1.2.dist-info/RECORD,,