Fix for Unused local variable

Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>

test/cluster/test_cluster_features.py

@@ -100,22 +100,24 @@ async def test_downgrade_after_partial_upgrade(manager: ManagerClient) -> None:
        in all nodes but one, then downgrading the upgraded nodes.
     """
     servers = await manager.running_servers()
-    cql, hosts = await manager.get_ready_cql(servers)
     upgrading_servers = servers[1:] if len(servers) > 0 else []
-    upgrading_hosts = hosts[1:] if len(hosts) > 0 else []
+
+    cql = manager.cql
 
     # Upgrade
-    for srv, host in zip(upgrading_servers, upgrading_hosts):
+    for srv in upgrading_servers:
         await change_support_for_test_feature_and_restart(manager, [srv], enable=True)
+        host = (await wait_for_cql_and_get_hosts(cql, [srv], time.time() + 60))[0]
         assert TEST_FEATURE_NAME in await get_supported_features(cql, host)
 
     # There is one node that is not upgraded. The feature should not be enabled.
-    for srv, host in zip(servers, hosts):
+    for srv in servers:
         assert TEST_FEATURE_NAME not in await get_enabled_features(cql, host)
 
     # Downgrade, in reverse order
-    for srv, host in zip(reversed(upgrading_servers), reversed(upgrading_hosts)):
+    for srv in upgrading_servers[::-1]:
         await change_support_for_test_feature_and_restart(manager, [srv], enable=False)
+        host = (await wait_for_cql_and_get_hosts(cql, [srv], time.time() + 60))[0]
         assert TEST_FEATURE_NAME not in await get_supported_features(cql, host)
 
 

test/cqlpy/run.py

@@ -182,7 +182,6 @@ def cleanup_all():
             sys.stdout.flush()
             shutil.copyfileobj(f, sys.stdout.buffer)
         f.close()
-    scylla_set = set()
     print(summary)
 
 # We run the cleanup_all() function on exit for any reason - successful finish