Update __init__.py:

- Add BaseDevice to __all__
- Add type str for register_parser
This commit is contained in:
i.shramko
2026-03-03 09:52:28 +03:00
parent 2a027ed736
commit 4ed34227d9

View File

@@ -6,7 +6,7 @@ device_registry = {}
def register_parser( def register_parser(
name: list[str], name: list[str] | str,
) -> Callable[[Type[BaseDevice]], Type[BaseDevice]]: ) -> Callable[[Type[BaseDevice]], Type[BaseDevice]]:
def wrapper(cls): def wrapper(cls):
name_list = [] name_list = []
@@ -23,4 +23,4 @@ def register_parser(
from . import models # noqa: E402, F401 from . import models # noqa: E402, F401
__all__ = ["register_parser", "device_registry"] __all__ = ["register_parser", "device_registry", "BaseDevice"]