Circuit-Forge/circuitforge-core
3
0
Fork 0
Code Issues 6 Pull requests Projects Releases 5 Packages Wiki Activity Actions

fix(resources): rename lambda var; convert asyncio.run test to async

Browse source
This commit is contained in:
pyr0ball 2026-03-30 20:41:03 -07:00
parent d60503f059
commit 6b239b76e3
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
circuitforge_core/resources/coordinator/lease_manager.py
View file

@ -65,7 +65,7 @@ class LeaseManager:
and lease.gpu_id == gpu_id and lease.gpu_id == gpu_id
and lease.priority > requester_priority 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] = [] selected: list[VRAMLease] = []
freed = 0 freed = 0
for candidate in candidates: for candidate in candidates:

8
tests/test_resources/test_lease_manager.py
View file

@ -1,4 +1,3 @@
import asyncio
import pytest import pytest
from circuitforge_core.resources.coordinator.lease_manager import LeaseManager 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 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 = LeaseManager()
mgr.register_gpu("heimdall", 0, 8192) mgr.register_gpu("heimdall", 0, 8192)
asyncio.run(mgr.try_grant("heimdall", 0, 3000, "a", 1)) await mgr.try_grant("heimdall", 0, 3000, "a", 1)
asyncio.run(mgr.try_grant("heimdall", 0, 2000, "b", 2)) await mgr.try_grant("heimdall", 0, 2000, "b", 2)
assert mgr.used_mb("heimdall", 0) == 5000 assert mgr.used_mb("heimdall", 0) == 5000