clean code

tests/test_repeat_init.py

@@ -16,13 +16,10 @@
 import multiprocessing
 
 import pytest
-# import time
 import fed
 import fed._private.compatible_utils as compatible_utils
 import ray
 
-# from fed.cleanup import _start_check_sending, push_to_sending
-
 
 @fed.remote
 class My:
@@ -48,7 +45,6 @@ def bar(self, li):
 
 def run(party):
  def _run():
- # assert fed.cleanup._sending_obj_refs_q is None
  compatible_utils.init_ray(address='local')
  fed.init(cluster=cluster, party=party)
 
@@ -64,7 +60,6 @@ def _run():
 
  fed.shutdown()
  ray.shutdown()
- # assert fed.cleanup._sending_obj_refs_q is None
 
  _run()
  _run()