diff --git a/tests/test_repeat_init.py b/tests/test_repeat_init.py index 3b5f55e..5e2d49f 100644 --- a/tests/test_repeat_init.py +++ b/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()