From ad494dd4c7a265fc160812f0bb8a07350d182248 Mon Sep 17 00:00:00 2001 From: Marijn Suijten Date: Sat, 26 Mar 2022 23:25:03 +0100 Subject: [PATCH] test_output: Fix type annotations --- test_output.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test_output.py b/test_output.py index b9cc56d..449bcf0 100644 --- a/test_output.py +++ b/test_output.py @@ -3,17 +3,17 @@ import asyncio -from omnikinverter import OmnikInverter +from omnikinverter import Device, Inverter, OmnikInverter -async def main(): +async def main() -> None: """Test.""" async with OmnikInverter( host="examples.com", source_type="javascript", ) as client: - inverter: OmnikInverter = await client.inverter() - device: OmnikInverter = await client.device() + inverter: Inverter = await client.inverter() + device: Device = await client.device() print(inverter) print() print("-- INVERTER --")