Circuit-Forge/peregrine
2
0
Fork 0
Code Issues 22 Pull requests 2 Projects Releases 6 Packages Wiki Activity Actions

refactor(scheduler): use module-level _get_gpus directly in __init__

Browse source
This commit is contained in:
pyr0ball 2026-03-15 04:01:01 -07:00
parent cceacdd371
commit fa780af2f1
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
scripts/task_scheduler.py
View file

@ -84,8 +84,7 @@ class TaskScheduler:
# Detect total GPU VRAM; fall back to unlimited (999) on CPU-only systems. # Detect total GPU VRAM; fall back to unlimited (999) on CPU-only systems.
# Uses module-level _get_gpus so tests can monkeypatch scripts.task_scheduler._get_gpus. # Uses module-level _get_gpus so tests can monkeypatch scripts.task_scheduler._get_gpus.
try: try:
from scripts import task_scheduler as _ts_mod gpus = _get_gpus()
gpus = _ts_mod._get_gpus()
self._available_vram: float = ( self._available_vram: float = (
sum(g["vram_total_gb"] for g in gpus) if gpus else 999.0 sum(g["vram_total_gb"] for g in gpus) if gpus else 999.0
) )