Add einops and other commonly used function in brianstate (#17)

chaobrain · Jun 14, 2024 · 2c77212 · 2c77212
1 parent 6d511c0
commit 2c77212
Show file tree

Hide file tree

Showing 7 changed files with 1,762 additions and 33 deletions.
diff --git a/brainunit/math/__init__.py b/brainunit/math/__init__.py
@@ -41,39 +41,41 @@
 from ._compat_numpy_linear_algebra import __all__ as _compat_linear_algebra_all
 from ._compat_numpy_misc import *
 from ._compat_numpy_misc import __all__ as _compat_misc_all
-# from ._compat_numpy import *
-# from ._compat_numpy import __all__ as _compat_numpy_all
+from ._einops import *
+from ._einops import __all__ as _einops_all
 from ._others import *
 from ._others import __all__ as _other_all
 
-__all__ = _compat_array_creation_all + \
-          _compat_array_manipulation_all + \
-          _compat_funcs_change_unit_all + \
-          _compat_funcs_keep_unit_all + \
-          _compat_funcs_accept_unitless_all + \
-          _compat_funcs_match_unit_all + \
-          _compat_funcs_remove_unit_all + \
-          _compat_get_attribute_all + \
-          _compat_funcs_bit_operation_all + \
-          _compat_funcs_logic_all + \
-          _compat_funcs_indexing_all + \
-          _compat_funcs_window_all + \
-          _compat_linear_algebra_all + \
-          _compat_misc_all + _other_all + \
-          _other_all
+__all__ = (_compat_array_creation_all +
+           _compat_array_manipulation_all +
+           _compat_funcs_change_unit_all +
+           _compat_funcs_keep_unit_all +
+           _compat_funcs_accept_unitless_all +
+           _compat_funcs_match_unit_all +
+           _compat_funcs_remove_unit_all +
+           _compat_get_attribute_all +
+           _compat_funcs_bit_operation_all +
+           _compat_funcs_logic_all +
+           _compat_funcs_indexing_all +
+           _compat_funcs_window_all +
+           _compat_linear_algebra_all +
+           _compat_misc_all + _other_all +
+           _other_all +
+           _einops_all)
 
-del _compat_array_creation_all, \
-  _compat_array_manipulation_all, \
-  _compat_funcs_change_unit_all, \
-  _compat_funcs_keep_unit_all, \
-  _compat_funcs_accept_unitless_all, \
-  _compat_funcs_match_unit_all, \
-  _compat_funcs_remove_unit_all, \
-  _compat_get_attribute_all, \
-  _compat_funcs_bit_operation_all, \
-  _compat_funcs_logic_all, \
-  _compat_funcs_indexing_all, \
-  _compat_funcs_window_all, \
-  _compat_linear_algebra_all, \
-  _compat_misc_all, \
-  _other_all
+del (_compat_array_creation_all,
+     _compat_array_manipulation_all,
+     _compat_funcs_change_unit_all,
+     _compat_funcs_keep_unit_all,
+     _compat_funcs_accept_unitless_all,
+     _compat_funcs_match_unit_all,
+     _compat_funcs_remove_unit_all,
+     _compat_get_attribute_all,
+     _compat_funcs_bit_operation_all,
+     _compat_funcs_logic_all,
+     _compat_funcs_indexing_all,
+     _compat_funcs_window_all,
+     _compat_linear_algebra_all,
+     _compat_misc_all,
+     _other_all,
+     _einops_all)