axsdb.units

Unit handling components, based on the Pint library.

Note

By default, Pint’s application registry is used.

Functions:

ensure_units(value, default_units[, convert])	Ensure that a value is wrapped in a Pint quantity container.
get_unit_registry()	Access the internal unit registry.
set_unit_registry(ureg)	Set internal unit registry.
xarray_to_quantity(da)	Converts a DataArray to a Quantity.

axsdb.units.ensure_units(value: Any, default_units: Unit, convert: bool = False) → Quantity

Ensure that a value is wrapped in a Pint quantity container.

Parameters:

• value – Checked value.

• default_units (pint.Unit) – Units to use to initialize the pint.Quantity if value is not a pint.Quantity.

• convert (bool, default: False) – If True, value will also be converted to default_units if it is a pint.Quantity.

Return type:

Converted value.

axsdb.units.get_unit_registry() → UnitRegistry

Access the internal unit registry. By default, the Pint application registry is returned.

axsdb.units.set_unit_registry(ureg: UnitRegistry | None) → None

Set internal unit registry.

Parameters:

ureg (pint.UnitRegistry or None) – New default unit registry. Set to None to reset to the default (the application registry).

axsdb.units.xarray_to_quantity(da: DataArray) → Quantity

Converts a DataArray to a Quantity. The array’s attrs metadata mapping must contain a units field.

Parameters:

da (DataArray) – DataArray instance which will be converted.

Returns:

The corresponding Pint quantity.

Return type:

quantity

Raises:

ValueError – If array attributes do not contain a units field.

Notes

This function can also be used on coordinate variables.