Merge branch 'main' of https://github.com/chaobrain/brainstate into s…

…ichao-patch-1

README.md

@@ -38,6 +38,6 @@ The official documentation is hosted on Read the Docs: [https://brainstate.readt
 
 ## See also the BDP ecosystem
 
-We are building the BDP ecosystem: https://ecosystem-for-brain-dynamics.readthedocs.io/
+We are building the Brain Dynamics Programming ecosystem: https://ecosystem-for-brain-dynamics.readthedocs.io/
 
 

brainstate/__init__.py

@@ -39,11 +39,11 @@
 
 __all__ = (
     [
-      'augment', 'compile', 'environ', 'event', 'functional',
-      'graph', 'init', 'mixin', 'nn', 'optim', 'random',
-      'surrogate', 'typing', 'util',
-      # deprecated
-      'transform',
+        'augment', 'compile', 'environ', 'event', 'functional',
+        'graph', 'init', 'mixin', 'nn', 'optim', 'random',
+        'surrogate', 'typing', 'util',
+        # deprecated
+        'transform',
     ] +
     _state_all
 )
@@ -56,9 +56,9 @@
 
 transform._deprecations = dict()
 for key in compile.__all__:
-  transform._deprecations[key] = (f'brainstate.transform.{key}', f'brainstate.compile.{key}', getattr(compile, key))
+    transform._deprecations[key] = (f'brainstate.transform.{key}', f'brainstate.compile.{key}', getattr(compile, key))
 for key in augment.__all__:
-  transform._deprecations[key] = (f'brainstate.transform.{key}', f'brainstate.augment.{key}', getattr(augment, key))
+    transform._deprecations[key] = (f'brainstate.transform.{key}', f'brainstate.augment.{key}', getattr(augment, key))
 transform.__getattr__ = deprecation_getattr('brainstate.transform', transform._deprecations)
 del deprecation_getattr