test_output: Fix type annotations

klaasnicolaas · Mar 26, 2022 · ad494dd · ad494dd
1 parent 315b608
commit ad494dd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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 --")