tweaks

Author: crusaderky

distributed/tests/test_scheduler.py

@@ -766,6 +766,7 @@ async def test_update_graph_culls(s, a, b):
 def test_io_loop(loop):
     s = Scheduler(loop=loop, port=0, dashboard_address=":0", validate=True)
     assert s.io_loop is loop
+    s.close()
 
 
 @gen_cluster(client=True)

distributed/tests/test_worker.py

@@ -1514,7 +1514,7 @@ async def test_interface_async(cleanup, loop, Worker):
 @pytest.mark.asyncio
 @pytest.mark.parametrize("Worker", [Worker, Nanny])
 async def test_protocol_from_scheduler_address(cleanup, Worker):
-    ucp = pytest.importorskip("ucp")
+    pytest.importorskip("ucp")
 
     async with Scheduler(protocol="ucx", port=0, dashboard_address=":0") as s:
         assert s.address.startswith("ucx://")