chore: Add types-pyyaml

Author: allenporter

.pre-commit-config.yaml

@@ -42,7 +42,7 @@ repos:
         hooks:
             -   id: mypy
                 exclude: cli.py
-                additional_dependencies: [ "types-paho-mqtt" ]
+                additional_dependencies: [ "types-paho-mqtt", "types-pyyaml" ]
     -   repo: https://github.com/alessandrojcm/commitlint-pre-commit-hook
         rev: v9.23.0
         hooks:

tests/devices/traits/v1/fixtures.py

@@ -118,9 +118,7 @@ def mock_app_get_init_status_fixture(device_info: HomeDataDevice, products: list
     product = next(filter(lambda product: product.id == device_info.product_id, products))
     if not product:
         raise ValueError(f"Product {device_info.product_id} not found")
-    device_info_data = mock_data.DEVICE_INFO.get(product.model)
-    if device_info_data is None:
-        return mock_data.APP_GET_INIT_STATUS
+    device_info_data = mock_data.DEVICE_INFO.get(product.model, {})
     return {
         **mock_data.APP_GET_INIT_STATUS,
         **device_info_data,