From 6b239b76e36650cb0837140c652512b0a4896f63 Mon Sep 17 00:00:00 2001 From: pyr0ball Date: Mon, 30 Mar 2026 20:41:03 -0700 Subject: [PATCH] fix(resources): rename lambda var; convert asyncio.run test to async --- circuitforge_core/resources/coordinator/lease_manager.py | 2 +- tests/test_resources/test_lease_manager.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/circuitforge_core/resources/coordinator/lease_manager.py b/circuitforge_core/resources/coordinator/lease_manager.py index f16259a..652b64b 100644 --- a/circuitforge_core/resources/coordinator/lease_manager.py +++ b/circuitforge_core/resources/coordinator/lease_manager.py @@ -65,7 +65,7 @@ class LeaseManager: and lease.gpu_id == gpu_id and lease.priority > requester_priority ] - candidates.sort(key=lambda l: l.priority, reverse=True) + candidates.sort(key=lambda lease: lease.priority, reverse=True) selected: list[VRAMLease] = [] freed = 0 for candidate in candidates: diff --git a/tests/test_resources/test_lease_manager.py b/tests/test_resources/test_lease_manager.py index 0b2385c..cede687 100644 --- a/tests/test_resources/test_lease_manager.py +++ b/tests/test_resources/test_lease_manager.py @@ -1,4 +1,3 @@ -import asyncio import pytest from circuitforge_core.resources.coordinator.lease_manager import LeaseManager @@ -77,9 +76,10 @@ def test_register_gpu_sets_total(mgr): assert mgr.gpu_total_mb("heimdall", 0) == 8192 -def test_used_mb_tracks_grants(): +@pytest.mark.asyncio +async def test_used_mb_tracks_grants(): mgr = LeaseManager() mgr.register_gpu("heimdall", 0, 8192) - asyncio.run(mgr.try_grant("heimdall", 0, 3000, "a", 1)) - asyncio.run(mgr.try_grant("heimdall", 0, 2000, "b", 2)) + await mgr.try_grant("heimdall", 0, 3000, "a", 1) + await mgr.try_grant("heimdall", 0, 2000, "b", 2) assert mgr.used_mb("heimdall", 0) == 5000