🚚 resource(unitsystems): move private module (#366)

* 🚚 resource(unitsystems): move private module * ⬆️ dep-bump(plum): bumpt bugfix version * 💚 ci(mypy): pin plum for pre-commit Signed-off-by: Nathaniel Starkman <[email protected]>
GalacticDynamics · Jan 15, 2025 · c165b3e · c165b3e
1 parent a9e0afb
commit c165b3e
Show file tree

Hide file tree

Showing 16 changed files with 18 additions and 17 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -79,7 +79,7 @@ repos:
       - id: mypy
         files: src
         additional_dependencies:
-          - plum-dispatch
+          - plum-dispatch>=2.5.6
 
   - repo: https://github.com/codespell-project/codespell
     rev: "v2.3.0"

diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,7 @@
     "jaxlib>0.4.32",
     "jaxtyping>=0.2.34",
     "optional-dependencies>=0.3.2",
-    "plum-dispatch>=2.5.5",
+    "plum-dispatch>=2.5.6",
     "quax>=0.0.5",
     "quaxed>=0.7.1",
     "xmmutablemap>=0.1",

diff --git a/src/unxt/_src/quantity/base.py b/src/unxt/_src/quantity/base.py
@@ -965,7 +965,8 @@ def from_(
 # -----------------------------------------------
 # Promotion rules
 
-add_promotion_rule(AbstractQuantity, AbstractQuantity, AbstractQuantity)  # type: ignore[no-untyped-call]
+add_promotion_rule(AbstractQuantity, AbstractQuantity, AbstractQuantity)
+
 
 # ===============================================================
 # Support for ``at``.

diff --git a/src/unxt/_src/quantity/register_api.py b/src/unxt/_src/quantity/register_api.py
@@ -14,7 +14,7 @@
 from unxt._src.dimensions import AbstractDimension, dimension_of
 from unxt._src.units import unit
 from unxt._src.units.api import AbstractUnits
-from unxt._src.units.system import AbstractUnitSystem
+from unxt._src.unitsystems import AbstractUnitSystem
 
 # ===================================================================
 # Get dimensions

diff --git a/src/unxt/_src/units/system/__init__.py → src/unxt/_src/unitsystems/__init__.py b/src/unxt/_src/units/system/__init__.py → src/unxt/_src/unitsystems/__init__.py
diff --git a/src/unxt/_src/units/system/base.py → src/unxt/_src/unitsystems/base.py b/src/unxt/_src/units/system/base.py → src/unxt/_src/unitsystems/base.py
diff --git a/src/unxt/_src/units/system/builtin.py → src/unxt/_src/unitsystems/builtin.py b/src/unxt/_src/units/system/builtin.py → src/unxt/_src/unitsystems/builtin.py
@@ -78,7 +78,7 @@ class SIUnitSystem(SingletonMixin, AbstractUnitSystem):
     Examples
     --------
     >>> import unxt as u
-    >>> from unxt._src.units.system.builtin import SIUnitSystem
+    >>> from unxt._src.unitsystems.builtin import SIUnitSystem
     >>> usys = SIUnitSystem(
     ...     length=u.unit("meter"),
     ...     time=u.unit("second"),
@@ -136,7 +136,7 @@ class CGSUnitSystem(SingletonMixin, AbstractUnitSystem):
     Examples
     --------
     >>> import unxt as u
-    >>> from unxt._src.units.system.builtin import CGSUnitSystem
+    >>> from unxt._src.unitsystems.builtin import CGSUnitSystem
     >>> usys = CGSUnitSystem(
     ...     length=u.unit("centimeter"),
     ...     time=u.unit("second"),

diff --git a/...t/_src/units/system/builtin_dimensions.py → ...xt/_src/unitsystems/builtin_dimensions.py b/...t/_src/units/system/builtin_dimensions.py → ...xt/_src/unitsystems/builtin_dimensions.py
diff --git a/src/unxt/_src/units/system/compare.py → src/unxt/_src/unitsystems/compare.py b/src/unxt/_src/units/system/compare.py → src/unxt/_src/unitsystems/compare.py
diff --git a/src/unxt/_src/units/system/core.py → src/unxt/_src/unitsystems/core.py b/src/unxt/_src/units/system/core.py → src/unxt/_src/unitsystems/core.py
diff --git a/src/unxt/_src/units/system/flags.py → src/unxt/_src/unitsystems/flags.py b/src/unxt/_src/units/system/flags.py → src/unxt/_src/unitsystems/flags.py
diff --git a/src/unxt/_src/units/system/realizations.py → src/unxt/_src/unitsystems/realizations.py b/src/unxt/_src/units/system/realizations.py → src/unxt/_src/unitsystems/realizations.py
diff --git a/src/unxt/_src/units/system/utils.py → src/unxt/_src/unitsystems/utils.py b/src/unxt/_src/units/system/utils.py → src/unxt/_src/unitsystems/utils.py
@@ -28,7 +28,7 @@ def get_dimension_name(pt: str, /) -> str:
 
     Examples
     --------
-    >>> from unxt._src.units.system.utils import get_dimension_name
+    >>> from unxt._src.unitsystems.utils import get_dimension_name
 
     >>> get_dimension_name("length")
     'length'
@@ -58,7 +58,7 @@ def get_dimension_name(pt: AbstractDimension, /) -> str:
     Examples
     --------
     >>> import unxt as u
-    >>> from unxt._src.units.system.utils import get_dimension_name
+    >>> from unxt._src.unitsystems.utils import get_dimension_name
 
     >>> get_dimension_name(u.dimension("length"))
     'length'
@@ -84,7 +84,7 @@ def get_dimension_name(pt: Unit, /) -> str:
     Examples
     --------
     >>> import unxt as u
-    >>> from unxt._src.units.system.utils import get_dimension_name
+    >>> from unxt._src.unitsystems.utils import get_dimension_name
 
     >>> get_dimension_name(u.unit("km"))
     'length'

diff --git a/src/unxt/unitsystems.py b/src/unxt/unitsystems.py
@@ -46,7 +46,7 @@
 from .setup_package import RUNTIME_TYPECHECKER
 
 with install_import_hook("unxt.unitsystems", RUNTIME_TYPECHECKER):
-    from ._src.units.system import (
+    from ._src.unitsystems import (
         NAMED_UNIT_SYSTEMS,
         UNITSYSTEMS_REGISTRY,
         AbstractUnitSystem,

diff --git a/tests/unit/test_unitsystems.py b/tests/unit/test_unitsystems.py
@@ -11,7 +11,7 @@
 import pytest
 
 from unxt import dimension, unit, unitsystems
-from unxt._src.units.system.base import _UNITSYSTEMS_REGISTRY
+from unxt._src.unitsystems.base import _UNITSYSTEMS_REGISTRY
 from unxt.unitsystems import (
     AbstractUnitSystem,
     AbstractUSysFlag,
@@ -28,7 +28,7 @@
 def clean_unitsystems_registry(monkeypatch):
     clean_registry = {}
     monkeypatch.setattr(
-        "unxt._src.units.system.base._UNITSYSTEMS_REGISTRY", clean_registry
+        "unxt._src.unitsystems.base._UNITSYSTEMS_REGISTRY", clean_registry
     )
     return clean_registry
 

diff --git a/uv.lock b/uv.lock