diff --git a/eitprocessing/datahandling/eitdata.py b/eitprocessing/datahandling/eitdata.py index 0af2f2133..6b4930ab6 100644 --- a/eitprocessing/datahandling/eitdata.py +++ b/eitprocessing/datahandling/eitdata.py @@ -2,12 +2,11 @@ import warnings from dataclasses import InitVar, dataclass, field -from enum import auto +from enum import Enum from pathlib import Path from typing import TYPE_CHECKING, Any, TypeVar import numpy as np -from strenum import LowercaseStrEnum # TODO: EOL 3.10: replace with native StrEnum from eitprocessing.datahandling import DataContainer from eitprocessing.datahandling.continuousdata import ContinuousData @@ -173,12 +172,28 @@ def calculate_global_impedance(self) -> np.ndarray: return np.nansum(self.pixel_impedance, axis=(1, 2)) -class Vendor(LowercaseStrEnum): - """Enum indicating the vendor (manufacturer) of the source EIT device.""" +class Vendor(Enum): + """Enum indicating the vendor (manufacturer) of the source EIT device. + + The enum values are all lowercase strings. For some manufacturers, multiple ways of wrinting are provided mapping to + the same value, to prevent confusion over conversion of special characters. The "simulated" vendor is provided to + indicate simulated data. + """ + + DRAEGER = "draeger" + """Dräger (PulmoVista V500)""" + + TIMPEL = "timpel" + """Timpel (Enlight 2100)""" + + SENTEC = "sentec" + """Sentec (Lumon)""" - DRAEGER = auto() - TIMPEL = auto() - SENTEC = auto() DRAGER = DRAEGER - DRÄGER = DRAEGER # noqa: PLC2401 - SIMULATED = auto() + """Synonym of DRAEGER""" + + DRÄGER = DRAEGER # noqa: PIE796, PLC2401 + """Synonym of DRAEGER""" + + SIMULATED = "simulated" + """Simulated data""" diff --git a/pyproject.toml b/pyproject.toml index f807f3562..d47ef0436 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,7 +40,6 @@ dependencies = [ "matplotlib>=3.10", "numpy >= 1.24.2", "scipy >= 1.10.1", - "strenum >= 0.4.10", "anytree >= 2.12.1 ", "typing_extensions", "pyyaml",