upgrade __init__.py

chaobrain · Jun 8, 2024 · 98417c1 · 98417c1
1 parent 64283c0
commit 98417c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/brainunit/__init__.py b/brainunit/__init__.py
@@ -15,6 +15,7 @@
 
 __version__ = "0.0.1"
 
+from . import math
 from ._base import *
 from ._base import __all__ as _base_all
 from ._unit_common import *
@@ -24,5 +25,5 @@
 from ._unit_shortcuts import *
 from ._unit_shortcuts import __all__ as _std_units_all
 
-__all__ = _common_all + _std_units_all + _constants_all + _base_all
+__all__ = ['math'] + _common_all + _std_units_all + _constants_all + _base_all
 del _common_all, _std_units_all, _constants_all, _base_all