polish

Author: Bilal Al

splitio/push/manager.py

@@ -349,6 +349,7 @@ async def stop(self, blocking=False):
 
         if self._token_task:
             self._token_task.cancel()
+            self._token_task = None
 
         if blocking:
             await self._stop_current_conn()

splitio/sync/synchronizer.py

@@ -614,7 +614,7 @@ async def sync_all(self, max_retry_attempts=_SYNC_ALL_NO_RETRIES):
                         break
                 how_long = self._backoff.get()
                 if not self._shutdown:
-                    time.sleep(how_long)
+                    await asyncio.sleep(how_long)
 
         _LOGGER.error("Could not correctly synchronize feature flags and segments after %d attempts.", retry_attempts)
 
@@ -838,7 +838,6 @@ async def stop_periodic_data_recording(self, blocking):
             asyncio.get_running_loop().create_task(self._stop_periodic_data_recording)
 
 
-
 class LocalhostSynchronizerBase(BaseSynchronizer):
     """LocalhostSynchronizer base."""