Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
remove math module
Browse files Browse the repository at this point in the history
chaoming0625 committed Jun 14, 2024
1 parent aea868f commit daf8447
Showing 10 changed files with 6 additions and 1,814 deletions.
9 changes: 4 additions & 5 deletions brainstate/__init__.py
Original file line number Diff line number Diff line change
@@ -20,17 +20,16 @@
__version__ = "0.0.1"

from . import environ
from . import math
from . import functional
from . import init
from . import mixin
from . import nn
from . import optim
from . import random
from . import surrogate
from . import transform
from . import typing
from . import util
from . import surrogate
from . import functional
from . import init
from ._module import *
from ._module import __all__ as _module_all
from ._state import *
@@ -39,7 +38,7 @@
__all__ = (
['environ', 'share', 'nn', 'optim', 'random',
'surrogate', 'functional', 'init',
'mixin', 'math', 'transform', 'util', 'typing'] +
'mixin', 'transform', 'util', 'typing'] +
_module_all + _state_all
)
del _module_all, _state_all
5 changes: 2 additions & 3 deletions brainstate/_module.py
Original file line number Diff line number Diff line change
@@ -46,7 +46,6 @@
"""

import inspect
import math
import numbers
from collections import namedtuple
@@ -58,12 +57,12 @@
import numpy as np

from . import environ
from ._utils import set_module_as
from ._state import State, StateDictManager, visible_state_dict
from .util import unique_name, DictManager, get_unique_name, DotDict
from ._utils import set_module_as
from .math import get_dtype
from .mixin import Mixin, Mode, DelayedInit, AllOfTypes, Batching, UpdateReturn
from .transform._jit_error import jit_error
from .util import unique_name, DictManager, get_unique_name

Shape = Union[int, Sequence[int]]
PyTree = Any
21 changes: 0 additions & 21 deletions brainstate/math/__init__.py

This file was deleted.

787 changes: 0 additions & 787 deletions brainstate/math/_einops.py

This file was deleted.

169 changes: 0 additions & 169 deletions brainstate/math/_einops_parsing.py

This file was deleted.

126 changes: 0 additions & 126 deletions brainstate/math/_einops_parsing_test.py

This file was deleted.

346 changes: 0 additions & 346 deletions brainstate/math/_einops_test.py

This file was deleted.

298 changes: 0 additions & 298 deletions brainstate/math/_misc.py

This file was deleted.

58 changes: 0 additions & 58 deletions brainstate/math/_misc_test.py

This file was deleted.

1 change: 0 additions & 1 deletion brainstate/util.py
Original file line number Diff line number Diff line change
@@ -26,7 +26,6 @@

from ._utils import set_module_as


__all__ = [
'unique_name',
'clear_buffer_memory',

0 comments on commit daf8447

Please sign in to comment.