diff --git a/labgrid/driver/power/tplink.py b/labgrid/driver/power/tplink.py index 6a538e776..e556a8732 100644 --- a/labgrid/driver/power/tplink.py +++ b/labgrid/driver/power/tplink.py @@ -1,14 +1,14 @@ """ Tested with TP Link KP303, and should be compatible with any strip supported by kasa """ import asyncio -from kasa import SmartStrip +from kasa.iot import IotStrip async def _power_set(host, port, index, value): """We embed the coroutines in an `async` function to minimise calls to `asyncio.run`""" assert port is None index = int(index) - strip = SmartStrip(host) + strip = IotStrip(host) await strip.update() assert ( len(strip.children) > index @@ -26,7 +26,7 @@ def power_set(host, port, index, value): def power_get(host, port, index): assert port is None index = int(index) - strip = SmartStrip(host) + strip = IotStrip(host) asyncio.run(strip.update()) assert ( len(strip.children) > index