Correct assertion for seconds in GenericWait.

diff --git a/controlpi_plugins/wait.py b/controlpi_plugins/wait.py
index 01861db8dedd29d97c153035f8fd50ea7597993d..be0c9e04e9d97ebfcdfb60e7dea2c094f2db602a 100644 (file)
--- a/controlpi_plugins/wait.py
+++ b/controlpi_plugins/wait.py
@@ -135,7 +135,8 @@ class GenericWait(BasePlugin):
     async def wait(self, message: Message) -> None:
         """Wait given time and send "finished" event with given "id"."""
         async def wait_coroutine():
-            assert isinstance(message['seconds'], float)
+            assert (isinstance(message['seconds'], float) or
+                    isinstance(message['seconds'], int))
             await asyncio.sleep(message['seconds'])
             await self.bus.send(Message(self.name, {'event': 'finished',
                                                     'id': message['id']}))