move units

tlambert03 · Jun 28, 2023 · 1a35d39 · 1a35d39
1 parent f5ee006
commit 1a35d39
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/src/ome_types2/_mixins/_base_type.py b/src/ome_types2/_mixins/_base_type.py
@@ -6,6 +6,8 @@
 
 from pydantic import BaseModel, validator
 
+from ome_types2.units import ureg
+
 if TYPE_CHECKING:
     import pint
 
@@ -129,7 +131,6 @@ def validate_id(cls, value: Any) -> Any:
 
 def _quantity_property(field_name: str) -> property:
     """Create property that returns a ``pint.Quantity`` combining value and unit."""
-    from ome_types2._units import ureg
 
     def quantity(self: Any) -> Optional["pint.Quantity"]:
         value = getattr(self, field_name)

diff --git a/src/ome_types2/_units.py → src/ome_types2/units.py b/src/ome_types2/_units.py → src/ome_types2/units.py
diff --git a/tests/v2/test_units_2.py b/tests/v2/test_units_2.py
@@ -2,9 +2,9 @@
 from pint import DimensionalityError
 from pydantic import ValidationError
 
-from ome_types2._units import ureg
 from ome_types2.model import simple_types
 from ome_types2.model.ome_2016_06 import Channel, Laser, Plane
+from ome_types2.units import ureg
 
 
 def test_quantity_math():