diff --git a/tests/golden/harness.py b/tests/golden/harness.py index 910487e..982e268 100644 --- a/tests/golden/harness.py +++ b/tests/golden/harness.py @@ -17,7 +17,7 @@ import json from dataclasses import dataclass from pathlib import Path -from typing import Any, Literal +from typing import Any, Literal, cast from flipperzero_mcp.transport.base import FlipperTransport @@ -54,7 +54,7 @@ def from_json(cls, obj: dict[str, str]) -> Event: direction = obj["dir"] if direction not in ("tx", "rx"): raise ValueError(f"invalid event direction: {direction!r}") - return cls(dir=direction, data=base64.b64decode(obj["b64"])) + return cls(dir=cast("Direction", direction), data=base64.b64decode(obj["b64"])) @dataclass