From: Benjamin Braatz <bb@bbraatz.eu>
Date: Mon, 22 Mar 2021 22:01:02 +0000 (+0100)
Subject: Correct assertion for seconds in GenericWait.
X-Git-Tag: v0.3.0~39
X-Git-Url: http://git.graph-it.com/?a=commitdiff_plain;h=5527584c7f565c488e3caac7ecd46790733d7154;p=graphit%2Fcontrolpi.git

Correct assertion for seconds in GenericWait.
---

diff --git a/controlpi_plugins/wait.py b/controlpi_plugins/wait.py
index 01861db..be0c9e0 100644
--- 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']}))