Async rollback

src/api.py

@@ -79,13 +79,13 @@ def after_close(self):
         pass
 
     @final
-    async def on_message(self, message):  # do not override
+    def on_message(self, message):  # do not override
         if self.last_message_time is None or time.time() - self.last_message_time >= 0.045:
             self.last_message_time = time.time()
-            await self.send_message(message)
+            self.send_message(message)
 
     @abstractmethod
-    async def send_message(self, message):
+    def send_message(self, message):
         raise NotImplementedError
 
 

src/bots.py

@@ -4,7 +4,7 @@
 
 
 class PongBot(BaseHandler):
-    async def send_message(self, message):
+    def send_message(self, message):
         data = json.loads(message)
         player = data['playerId']
         state = data['state']
@@ -20,11 +20,11 @@ async def send_message(self, message):
             else:
                 move = -1
 
-        await self.write_message(json.dumps({'move': move, 'start': 1}))
+        self.write_message(json.dumps({'move': move, 'start': 1}))
 
 
 class FlappybirdBot(BaseHandler):
-    async def send_message(self, message):
+    def send_message(self, message):
         state = json.loads(message)['state']
         jump = 0
 
@@ -38,11 +38,11 @@ async def send_message(self, message):
             if state['birdY'] > nearest_obstacle['centerGapY']:
                 jump = 1
 
-        await self.write_message(json.dumps({'jump': jump}))
+        self.write_message(json.dumps({'jump': jump}))
 
 
 class SkijumpBot(BaseHandler):
-    async def send_message(self, message):
+    def send_message(self, message):
         state = json.loads(message)['state']
 
         space = 0
@@ -58,4 +58,4 @@ async def send_message(self, message):
         if state['jumperInclineRad'] < 0.7:
             up = 1
 
-        await self.write_message(json.dumps({'space': space, 'up': up, 'down': down}))
+        self.write_message(json.dumps({'space': space, 'up': up, 'down': down}))

src/handlers.py

@@ -18,7 +18,7 @@ def initialize(
     def after_close(self):
         self.agent.states.clear()
 
-    async def send_message(self, message):
+    def send_message(self, message):
         data = json.loads(message)
         action = self.agent.return_prediction(data)
-        await self.write_message(json.dumps(action))
+        self.write_message(json.dumps(action))