lint

Signed-off-by: Qing Wang <kingchin1218@gmail.com>

fed/_private/global_context.py

@@ -14,6 +14,7 @@
 
 from fed.cleanup import CleanupManager
 
+
 class GlobalContext:
     def __init__(self) -> None:
         self._seq_count = 0

fed/cleanup.py

@@ -18,7 +18,6 @@
 import threading
 import time
 from collections import deque
-from fed.utils import LockGuard
 
 import ray
 
@@ -54,7 +53,6 @@ def _main_thread_monitor():
         self._monitor_thread.start()
         logger.info('Start check sending monitor thread.')
 
-
     def _stop_gracefully(self):
         assert self._check_send_thread is not None
         self._notify_to_exit()
@@ -68,7 +66,6 @@ def _notify_to_exit(self):
     def push_to_sending(self, obj_ref: ray.ObjectRef):
         self._sending_obj_refs_q.append(obj_ref)
 
-
     def _check_sending_objs(self):
         def _signal_exit():
             os.kill(os.getpid(), signal.SIGTERM)

tests/test_repeat_init.py

@@ -16,7 +16,7 @@
 import multiprocessing
 
 import pytest
-import time
+# import time
 import fed
 import fed._private.compatible_utils as compatible_utils
 import ray
@@ -51,13 +51,6 @@ def _run():
         # assert fed.cleanup._sending_obj_refs_q is None
         compatible_utils.init_ray(address='local')
         fed.init(cluster=cluster, party=party)
-        # _start_check_sending()
-        # time.sleep(0.5)
-        # assert fed.cleanup._sending_obj_refs_q is not None
-        # push_to_sending(True)
-        # # Slightly longer than the queue polling
-        # time.sleep(0.6)
-        # assert fed.cleanup._sending_obj_refs_q is None
 
         my1 = My.party("alice").remote()
         my2 = My.party("bob").remote()