mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-08-09 06:58:41 -04:00
Merge verified Odysseus fixes
This commit is contained in:
@@ -1,7 +1,12 @@
|
||||
from services.hwfit.image_models import rank_image_models, IMAGE_MODEL_REGISTRY
|
||||
from services.hwfit import image_models
|
||||
|
||||
rank_image_models = image_models.rank_image_models
|
||||
IMAGE_MODEL_REGISTRY = image_models.IMAGE_MODEL_REGISTRY
|
||||
|
||||
|
||||
def test_rank_image_models_handles_non_dict_system():
|
||||
def test_rank_image_models_handles_non_dict_system(monkeypatch):
|
||||
monkeypatch.setattr(image_models, "_fetch_hf_image_collection_models", lambda: [])
|
||||
monkeypatch.setattr(image_models, "_discover_quant_repos", lambda *a, **k: {})
|
||||
# `system` is the detected-hardware dict; if detection failed and returned
|
||||
# None (or a non-dict), system.get(...) raised AttributeError. Treat a
|
||||
# non-dict system as "unknown hardware" (no GPU) rather than crashing.
|
||||
|
||||
Reference in New Issue
Block a user