diff options
Diffstat (limited to 'pint/facets/measurement/registry.py')
-rw-r--r-- | pint/facets/measurement/registry.py | 21 |
1 files changed, 16 insertions, 5 deletions
diff --git a/pint/facets/measurement/registry.py b/pint/facets/measurement/registry.py index 0fc4391..4a3e878 100644 --- a/pint/facets/measurement/registry.py +++ b/pint/facets/measurement/registry.py @@ -9,15 +9,17 @@ from __future__ import annotations -from ...compat import ufloat +from typing import Generic, Any + +from ...compat import ufloat, TypeAlias from ...util import create_class_with_registry -from ..plain import PlainRegistry -from .objects import MeasurementQuantity +from ..plain import GenericPlainRegistry, QuantityT, UnitT from . import objects -class MeasurementRegistry(PlainRegistry): - Quantity = MeasurementQuantity +class GenericMeasurementRegistry( + Generic[QuantityT, UnitT], GenericPlainRegistry[QuantityT, UnitT] +): Measurement = objects.Measurement def _init_dynamic_classes(self) -> None: @@ -34,3 +36,12 @@ class MeasurementRegistry(PlainRegistry): ) self.Measurement = no_uncertainties + + +class MeasurementRegistry( + GenericMeasurementRegistry[ + objects.MeasurementQuantity[Any], objects.MeasurementUnit + ] +): + Quantity: TypeAlias = objects.MeasurementQuantity[Any] + Unit: TypeAlias = objects.MeasurementUnit |