fix imports

chaobrain · Jun 12, 2024 · 859acc5 · 859acc5
1 parent 0d322d3
commit 859acc5
Show file tree

Hide file tree

Showing 16 changed files with 20 additions and 19 deletions.
diff --git a/brainunit/math/_misc.py → brainunit/_misc.py b/brainunit/math/_misc.py → brainunit/_misc.py
diff --git a/brainunit/math/_compat_numpy_array_creation.py b/brainunit/math/_compat_numpy_array_creation.py
@@ -21,7 +21,7 @@
 import numpy as np
 from jax import Array
 
-from ._misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import (
   DIMENSIONLESS,
   Quantity,

diff --git a/brainunit/math/_compat_numpy_array_manipulation.py b/brainunit/math/_compat_numpy_array_manipulation.py
@@ -20,8 +20,9 @@
 import jax.numpy as jnp
 from jax import Array
 
-from ._misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import Quantity
+
 __all__ = [
   # array manipulation
   'reshape', 'moveaxis', 'transpose', 'swapaxes', 'row_stack',

diff --git a/brainunit/math/_compat_numpy_funcs_accept_unitless.py b/brainunit/math/_compat_numpy_funcs_accept_unitless.py
@@ -19,7 +19,7 @@
 import jax.numpy as jnp
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import (Quantity, fail_for_dimension_mismatch, )
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_funcs_bit_operation.py b/brainunit/math/_compat_numpy_funcs_bit_operation.py
@@ -19,7 +19,7 @@
 import numpy as np
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import Quantity
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_funcs_change_unit.py b/brainunit/math/_compat_numpy_funcs_change_unit.py
@@ -19,10 +19,10 @@
 import jax.numpy as jnp
 import numpy as np
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from ._compat_numpy_get_attribute import isscalar
 from .._base import (DIMENSIONLESS,
-                     Quantity,)
+                     Quantity, )
 from .._base import _return_check_unitless
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_funcs_indexing.py b/brainunit/math/_compat_numpy_funcs_indexing.py
@@ -18,10 +18,10 @@
 import jax.numpy as jnp
 import numpy as np
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import (Quantity,
                      fail_for_dimension_mismatch,
-                     is_unitless,)
+                     is_unitless, )
 
 __all__ = [
 

diff --git a/brainunit/math/_compat_numpy_funcs_keep_unit.py b/brainunit/math/_compat_numpy_funcs_keep_unit.py
@@ -18,7 +18,7 @@
 import jax.numpy as jnp
 import numpy as np
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import Quantity
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_funcs_logic.py b/brainunit/math/_compat_numpy_funcs_logic.py
@@ -19,9 +19,9 @@
 import numpy as np
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import (Quantity,
-                     fail_for_dimension_mismatch,)
+                     fail_for_dimension_mismatch, )
 
 __all__ = [
   # logic funcs (unary)

diff --git a/brainunit/math/_compat_numpy_funcs_match_unit.py b/brainunit/math/_compat_numpy_funcs_match_unit.py
@@ -19,9 +19,9 @@
 import numpy as np
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import (Quantity,
-                     fail_for_dimension_mismatch,)
+                     fail_for_dimension_mismatch, )
 
 __all__ = [
   # math funcs match unit (binary)

diff --git a/brainunit/math/_compat_numpy_funcs_remove_unit.py b/brainunit/math/_compat_numpy_funcs_remove_unit.py
@@ -17,7 +17,7 @@
 import jax.numpy as jnp
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import Quantity
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_funcs_window.py b/brainunit/math/_compat_numpy_funcs_window.py
@@ -16,7 +16,7 @@
 import jax.numpy as jnp
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 
 __all__ = [
 

diff --git a/brainunit/math/_compat_numpy_get_attribute.py b/brainunit/math/_compat_numpy_get_attribute.py
@@ -18,7 +18,7 @@
 import jax.numpy as jnp
 import numpy as np
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from .._base import Quantity
 
 __all__ = [

diff --git a/brainunit/math/_compat_numpy_linear_algebra.py b/brainunit/math/_compat_numpy_linear_algebra.py
@@ -17,7 +17,7 @@
 import jax.numpy as jnp
 from jax import Array
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from ._compat_numpy_funcs_change_unit import funcs_change_unit_binary
 from ._compat_numpy_funcs_keep_unit import funcs_keep_unit_unary
 from .._base import Quantity

diff --git a/brainunit/math/_compat_numpy_misc.py b/brainunit/math/_compat_numpy_misc.py
@@ -22,7 +22,7 @@
 from jax import Array
 from jax._src.numpy.lax_numpy import _einsum
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from ._compat_numpy_array_manipulation import func_array_manipulation
 from ._compat_numpy_funcs_change_unit import funcs_change_unit_binary
 from ._compat_numpy_funcs_keep_unit import funcs_keep_unit_unary

diff --git a/brainunit/math/_others.py b/brainunit/math/_others.py
@@ -15,7 +15,7 @@
 
 import jax.numpy as jnp
 
-from _misc import set_module_as
+from brainunit._misc import set_module_as
 from ._compat_numpy_funcs_accept_unitless import funcs_only_accept_unitless_unary
 
 __all__ = [