Revert remaining test changes

rapidsai · Oct 3, 2024 · 5ea83e5 · 5ea83e5
1 parent 52ded8d
commit 5ea83e5
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/python/distributed-ucxx/distributed_ucxx/tests/test_ucxx.py b/python/distributed-ucxx/distributed_ucxx/tests/test_ucxx.py
@@ -85,9 +85,6 @@ async def test_comm_objs(ucxx_loop):
 
     assert comm.peer_address == serv_comm.local_address
 
-    await comm.close()
-    await serv_comm.close()
-
 
 @gen_test()
 async def test_ucxx_specific(ucxx_loop):
@@ -320,10 +317,7 @@ async def test_simple(
     ucxx_loop,
 ):
     async with LocalCluster(
-        protocol="ucxx",
-        n_workers=2,
-        threads_per_worker=2,
-        asynchronous=True,
+        protocol="ucxx", n_workers=2, threads_per_worker=2, asynchronous=True
     ) as cluster:
         async with Client(cluster, asynchronous=True) as client:
             assert cluster.scheduler_address.startswith("ucxx://")