brainstate 0.1.0.post20250503__py2.py3-none-any.whl → 0.1.1__py2.py3-none-any.whl

brainstate/functional/_spikes.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
 
 __all__ = [
     'spike_bitwise_or',

brainstate/graph/_graph_node.py

@@ -15,8 +15,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from __future__ import annotations
-
 from abc import ABCMeta
 from copy import deepcopy
 from typing import Any, Callable, Type, TypeVar, Tuple, TYPE_CHECKING, Mapping, Iterator, Sequence
@@ -210,7 +208,7 @@ class List(Node):
     def __len__(self):
         return len(vars(self))
 
-    def __add__(self, other: Sequence[A]) -> List[A]:
+    def __add__(self, other: Sequence[A]) -> 'List[A]':
         return List(list(self) + list(other))
 
     def append(self, value):

brainstate/graph/_graph_operation.py

@@ -18,8 +18,10 @@
 from __future__ import annotations
 
 import dataclasses
-from typing import (Any, Callable, Generic, Iterable, Iterator, Mapping, MutableMapping,
-                    Sequence, Type, TypeVar, Union, Hashable, Tuple, Dict, Optional, overload)
+from typing import (
+    Any, Callable, Generic, Iterable, Iterator, Mapping, MutableMapping,
+    Sequence, Type, TypeVar, Union, Hashable, Tuple, Dict, Optional, overload
+)
 
 import jax
 import numpy as np

brainstate/init/_base.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 from typing import Optional, Tuple
 
 import numpy as np

brainstate/init/_generic.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 from typing import Union, Callable, Optional, Sequence
 

brainstate/init/_random_inits.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 import math
 

brainstate/init/_regular_inits.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 import brainunit as u
 
 from brainstate import environ

brainstate/mixin.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from typing import (Sequence, Optional, TypeVar, _SpecialForm, _type_check, _remove_dups_flatten, _UnionGenericAlias)
 
 from brainstate.typing import PyTree

brainstate/nn/_collective_ops.py

@@ -13,13 +13,10 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 from collections import namedtuple
 from typing import Callable, TypeVar, Tuple, Any, Dict
 
 import jax
-from typing import Callable, TypeVar, Tuple, Any, Dict
 
 from brainstate._state import catch_new_states
 from brainstate._utils import set_module_as

brainstate/nn/_common.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from collections import defaultdict
 from typing import Any, Sequence, Hashable, Dict
 

brainstate/nn/_dyn_impl/_dynamics_neuron.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from typing import Callable, Optional
 
 import brainunit as u

brainstate/nn/_dyn_impl/_dynamics_synapse.py

@@ -15,7 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
 
 from typing import Optional, Callable
 

brainstate/nn/_dyn_impl/_inputs.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import annotations
 
 from typing import Union, Optional, Sequence, Callable
 

brainstate/nn/_dyn_impl/_rate_rnns.py

@@ -15,7 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
 
 from typing import Callable, Union
 

brainstate/nn/_dyn_impl/_readout.py

@@ -15,7 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
 
 import numbers
 from typing import Callable

brainstate/nn/_dynamics/_dynamics_base.py

@@ -32,7 +32,6 @@ For handling the delays:
 - ``DelayAccess``: The class for the delay access.
 
 """
-from __future__ import annotations
 
 from typing import Any, Dict, Callable, Hashable, Optional, Union, TypeVar, TYPE_CHECKING
 

brainstate/nn/_dynamics/_projection_base.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import annotations
 
 from typing import Union, Callable, Optional
 

brainstate/nn/_dynamics/_state_delay.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 import math
 import numbers
 from functools import partial

brainstate/nn/_dynamics/_synouts.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 import brainunit as u
 import jax.numpy as jnp
 

brainstate/nn/_elementwise/_dropout.py

@@ -14,8 +14,6 @@
 # ==============================================================================
 
 
-from __future__ import annotations
-
 from functools import partial
 from typing import Optional, Sequence
 

brainstate/nn/_elementwise/_elementwise.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from typing import Optional
 
 import brainunit as u

brainstate/nn/_event/_fixedprob_mv.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
 
 from typing import Union, Callable, Optional
 

brainstate/nn/_event/_linear_mv.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 from typing import Union, Callable, Optional
 
 import brainunit as u

brainstate/nn/_exp_euler.py

@@ -14,8 +14,6 @@
 # ==============================================================================
 
 
-from __future__ import annotations
-
 from typing import Callable
 
 import brainunit as u

brainstate/nn/_interaction/_conv.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 import collections.abc
 from typing import Callable, Tuple, Union, Sequence, Optional, TypeVar
 

brainstate/nn/_interaction/_embedding.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import annotations
 
 from typing import Optional, Callable, Union
 

brainstate/nn/_interaction/_linear.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from typing import Callable, Union, Optional
 
 import brainunit as u

brainstate/nn/_interaction/_normalizations.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 from typing import Callable, Union, Sequence, Optional, Any
 
 import jax

brainstate/nn/_interaction/_poolings.py

@@ -15,8 +15,6 @@
 
 # -*- coding: utf-8 -*-
 
-from __future__ import annotations
-
 import functools
 from typing import Sequence, Optional
 from typing import Union, Tuple, Callable, List

brainstate/nn/_module.py

@@ -25,7 +25,6 @@ The basic classes include:
 - ``Sequential``: The class for a sequential of modules, which update the modules sequentially.
 
 """
-from __future__ import annotations
 
 import warnings
 from typing import Sequence, Optional, Tuple, Union, TYPE_CHECKING, Callable

brainstate/nn/metrics.py

@@ -14,8 +14,6 @@
 # ==============================================================================
 
 
-from __future__ import annotations
-
 import typing as tp
 from dataclasses import dataclass
 from functools import partial

brainstate/optim/_base.py

@@ -14,8 +14,6 @@
 # ==============================================================================
 
 
-from __future__ import annotations
-
 from typing import Dict, Hashable
 
 from brainstate._state import State, StateDictManager

brainstate/optim/_lr_scheduler.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 from typing import Sequence, Union
 

brainstate/optim/_optax_optimizer.py

@@ -14,8 +14,6 @@
 # ==============================================================================
 
 
-from __future__ import annotations
-
 import importlib.util
 from typing import Hashable, Dict, Optional
 

brainstate/optim/_sgd_optimizer.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 import functools
 from typing import Union, Dict, Optional, Tuple, Any, TypeVar

brainstate/random/_rand_funs.py

@@ -15,7 +15,6 @@
 
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 from typing import Optional
 

brainstate/random/_rand_seed.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import annotations
 
 from contextlib import contextmanager
 from typing import Optional

brainstate/random/_rand_state.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 from functools import partial
 from operator import index

brainstate/surrogate.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 
 # -*- coding: utf-8 -*-
-from __future__ import annotations
 
 import jax
 import jax.numpy as jnp

brainstate/typing.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 import builtins
 import functools as ft
 import importlib

brainstate/util/_caller.py

@@ -15,8 +15,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from __future__ import annotations
-
 import dataclasses
 from typing import Any, TypeVar, Protocol, Generic
 
@@ -82,18 +80,18 @@ class CallableProxy:
     def __call__(self, *args, **kwargs):
         return self._callable(self._accessor, *args, **kwargs)
 
-    def __getattr__(self, name) -> CallableProxy:
+    def __getattr__(self, name) -> 'CallableProxy':
         return CallableProxy(self._callable, getattr(self._accessor, name))
 
-    def __getitem__(self, key) -> CallableProxy:
+    def __getitem__(self, key) -> 'CallableProxy':
         return CallableProxy(self._callable, self._accessor[key])
 
 
 class ApplyCaller(Protocol, Generic[A]):
-    def __getattr__(self, __name) -> ApplyCaller[A]:
+    def __getattr__(self, __name) -> 'ApplyCaller[A]':
         ...
 
-    def __getitem__(self, __name) -> ApplyCaller[A]:
+    def __getitem__(self, __name) -> 'ApplyCaller[A]':
         ...
 
     def __call__(self, *args, **kwargs) -> tuple[Any, A]:

brainstate/util/_others.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-from __future__ import annotations
-
 import copy
 import functools
 import gc