brainstate 0.1.0.post20250501__py2.py3-none-any.whl → 0.1.0.post20250503__py2.py3-none-any.whl

brainstate/_compatible_import.py

@@ -65,7 +65,7 @@ else:
             trace_ctx.set_axis_env(prev)
 
 if jax.__version_info__ < (0, 6, 0):
-    from jax.util import safe_map, safe_zip, unzip2, unzip3, wraps
+    from jax.util import safe_map, safe_zip, unzip2, wraps
 
 else:
     def safe_map(f, *args):
@@ -97,21 +97,6 @@ else:
         return tuple(xs), tuple(ys)
 
 
-    def unzip3(xyzs: Iterable[tuple[T1, T2, T3]]
-               ) -> tuple[tuple[T1, ...], tuple[T2, ...], tuple[T3, ...]]:
-        """Unzip sequence of length-3 tuples into three tuples."""
-        # Note: we deliberately don't use zip(*xyzs) because it is lazily evaluated,
-        # is too permissive about inputs, and does not guarantee a length-3 output.
-        xs: list[T1] = []
-        ys: list[T2] = []
-        zs: list[T3] = []
-        for x, y, z in xyzs:
-            xs.append(x)
-            ys.append(y)
-            zs.append(z)
-        return tuple(xs), tuple(ys), tuple(zs)
-
-
     def fun_name(fun: Callable):
         name = getattr(fun, "__name__", None)
         if name is not None:

{brainstate-0.1.0.post20250501.dist-info → brainstate-0.1.0.post20250503.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainstate
-Version: 0.1.0.post20250501
+Version: 0.1.0.post20250503
 Summary: A ``State``-based Transformation System for Program Compilation and Augmentation.
 Home-page: https://github.com/chaobrain/brainstate
 Author: BrainState Developers
@@ -48,7 +48,7 @@ Requires-Dist: jaxlib[tpu] ; extra == 'tpu'
 
 
 <p align="center">
-  	<img alt="Header image of brainstate." src="https://github.com/chaobrain/brainstate/blob/main/docs/_static/brainstate.png" width=40%>
+  	<img alt="Header image of brainstate." src="https://raw.githubusercontent.com/chaobrain/brainstate/main/docs/_static/brainstate.png" width=40%>
 </p> 
 
 
@@ -74,7 +74,11 @@ You can install ``brainstate`` via pip:
 pip install brainstate --upgrade
 ```
 
+Alternatively, you can install `BrainX`, which bundles `brainstate` with other compatible packages for a comprehensive brain modeling ecosystem:
 
+```bash
+pip install BrainX -U
+```
 
 ## Documentation
 

{brainstate-0.1.0.post20250501.dist-info → brainstate-0.1.0.post20250503.dist-info}/RECORD

@@ -1,5 +1,5 @@
 brainstate/__init__.py,sha256=AkZyyFkn4fB8g2aT6Rc2MO1xICPpUZuDtdze-eUQNc0,1496
-brainstate/_compatible_import.py,sha256=nEmjwX7aCH_kiXsVJXMjOd_SMj2etMiHxCF3z3H8GF0,5444
+brainstate/_compatible_import.py,sha256=7YGGnkoHMSV4hrTUTN0pTH8bmBCnu0L6E-eB8EdUokQ,4830
 brainstate/_state.py,sha256=Qvb6O3LFUcq_V8wG6GLAgkckImEwfzDi79JFrZ-lVWc,60753
 brainstate/_state_test.py,sha256=b6uvZdVRyC4n6-fYzmHNry1b-gJ6zE_kRSxGinqiHaw,1638
 brainstate/_utils.py,sha256=uJ6WWKq3yb05ZdktCQGLWOXsOJveL1H9pR7eev70Jes,1693
@@ -126,8 +126,8 @@ brainstate/util/_pretty_table.py,sha256=c3c2UH8hIZ-lCas_KL462kLgsnBc6pjJiMuapPi6
 brainstate/util/_scaling.py,sha256=pc_eM_SZVwkY65I4tJh1ODiHNCoEhsfFXl2zBK0PLAg,7562
 brainstate/util/_struct.py,sha256=RNQewdpyNcfntOJYefsDVWQjZW3RLsISKmTHpBhzyc8,17517
 brainstate/util/filter.py,sha256=blTktYNaNgCsuwv7xABjvbWsoi4Fozov6C2cXX-ta2g,14124
-brainstate-0.1.0.post20250501.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
-brainstate-0.1.0.post20250501.dist-info/METADATA,sha256=l6z_DHPZegmbLm2JoQT101SmHbULrVJ5yLn-Dd_NvhM,3655
-brainstate-0.1.0.post20250501.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-brainstate-0.1.0.post20250501.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
-brainstate-0.1.0.post20250501.dist-info/RECORD,,
+brainstate-0.1.0.post20250503.dist-info/LICENSE,sha256=VZe9u1jgUL2eCY6ZPOYgdb8KCblCHt8ECdbtJid6e1s,11550
+brainstate-0.1.0.post20250503.dist-info/METADATA,sha256=OSUkTgf1YBIYUojDxojQcil4Gwj5pC2pEa8URkbxm7Y,3848
+brainstate-0.1.0.post20250503.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+brainstate-0.1.0.post20250503.dist-info/top_level.txt,sha256=eQbGgKn0ptx7FDWuua0V0wr4K1VHi2iOUCYo3fUQBRA,11
+brainstate-0.1.0.post20250503.dist-info/RECORD,,