update imports

chaobrain · Aug 9, 2024 · 86c2738 · 86c2738
1 parent 09bfd9f
commit 86c2738
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 12 deletions.
diff --git a/brainunit/math/_einops_test.py b/brainunit/math/_einops_test.py
@@ -18,9 +18,9 @@
 import pytest
 
 import brainunit as bu
+from brainunit._base import assert_quantity
 from brainunit.math._einops import einrearrange, einreduce, einrepeat, _enumerate_directions
 from brainunit.math._einops_parsing import EinopsError
-from brainunit._base import assert_quantity
 
 REDUCTIONS = ("min", "max", "sum", "mean", "prod")
 

diff --git a/brainunit/math/_fun_accept_unitless.py b/brainunit/math/_fun_accept_unitless.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 from __future__ import annotations
+
 from typing import Union, Optional, Tuple, Any, Callable
 
 import jax

diff --git a/brainunit/math/_fun_array_creation.py b/brainunit/math/_fun_array_creation.py
@@ -25,7 +25,6 @@
 from .._base import (
   Quantity,
   Unit,
-  fail_for_dimension_mismatch,
   fail_for_unit_mismatch,
   get_unit,
   unit_scale_align_to_first,

diff --git a/brainunit/math/_fun_array_creation_test.py b/brainunit/math/_fun_array_creation_test.py
@@ -4,7 +4,7 @@
 
 import brainunit as bu
 import brainunit.math as bm
-from brainunit import second, meter, DimensionMismatchError
+from brainunit import second, meter
 from brainunit._base import assert_quantity
 
 fun_array_creation_given_shape = [

diff --git a/brainunit/math/_fun_change_unit.py b/brainunit/math/_fun_change_unit.py
@@ -21,7 +21,7 @@
 import jax.numpy as jnp
 
 from ._fun_array_creation import asarray
-from .._base import DIMENSIONLESS, UNITLESS, Quantity, remove_unitless
+from .._base import UNITLESS, Quantity, remove_unitless
 from .._misc import set_module_as
 
 __all__ = [

diff --git a/brainunit/math/_fun_remove_unit.py b/brainunit/math/_fun_remove_unit.py
@@ -19,7 +19,7 @@
 import jax
 import jax.numpy as jnp
 
-from .._base import Quantity, fail_for_dimension_mismatch, get_unit, fail_for_unit_mismatch, UNITLESS
+from .._base import Quantity, get_unit
 from .._misc import set_module_as
 
 __all__ = [

diff --git a/brainunit/math/_fun_remove_unit_test.py b/brainunit/math/_fun_remove_unit_test.py
@@ -7,7 +7,6 @@
 
 import brainunit as bu
 import brainunit.math as bm
-from brainunit import DimensionMismatchError
 from brainunit._base import assert_quantity
 
 fun_remove_unit_unary = [

diff --git a/brainunit/math/_misc.py b/brainunit/math/_misc.py
@@ -22,13 +22,10 @@
 import jax.numpy as jnp
 import numpy as np
 
-from .._base import (DIMENSIONLESS,
-UNITLESS,
-Unit,
+from .._base import (Unit,
                      Quantity,
-get_unit,
-                     is_unitless,
-                     get_dim)
+                     get_unit,
+                     is_unitless)
 from .._misc import set_module_as
 
 T = TypeVar("T")