diff --git a/oxi/interfaces/__init__.py b/oxi/interfaces/__init__.py index 4edb8b3..c5f11ea 100644 --- a/oxi/interfaces/__init__.py +++ b/oxi/interfaces/__init__.py @@ -6,7 +6,7 @@ device_registry = {} def register_parser( - name: list[str], + name: list[str] | str, ) -> Callable[[Type[BaseDevice]], Type[BaseDevice]]: def wrapper(cls): name_list = [] @@ -23,4 +23,4 @@ def register_parser( from . import models # noqa: E402, F401 -__all__ = ["register_parser", "device_registry"] +__all__ = ["register_parser", "device_registry", "BaseDevice"]