CalvinBackup/facefusion
1
0 Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-10-13 17:17:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

remove try block, make detect_graphic_devices stateless, cosmetics

Browse Source
This commit is contained in:
harisreedhar authored and henryruhs committed 2026-05-11 16:35:16 +02:00
1 parent 9e63832740
commit f475ec4bbc
4 files changed
+131 -138
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/test_api_metrics.py
+6 -6
View File
@@ -167,9 +167,9 @@ def test_get_metrics(test_client : TestClient) -> None:
assert metrics_response.status_code == 200
assert metrics_body.get('execution_devices')[0].get('driver_version') == '555.42'
assert metrics_body.get('execution_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_body.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_body.get('graphic_devices')[0].get('driver_version') == '555.42'
assert metrics_body.get('graphic_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_body.get('graphic_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_body.get('disks')[0].get('total').get('value') == 500
assert metrics_body.get('disks')[0].get('free').get('unit') == 'GB'
@@ -201,9 +201,9 @@ def test_websocket_metrics(test_client : TestClient) -> None:
]) as websocket:
metrics_set = websocket.receive_json()
assert metrics_set.get('execution_devices')[0].get('driver_version') == '555.42'
assert metrics_set.get('execution_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_set.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_set.get('graphic_devices')[0].get('driver_version') == '555.42'
assert metrics_set.get('graphic_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_set.get('graphic_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_set.get('disks')[0].get('total').get('value') == 500
assert metrics_set.get('disks')[0].get('free').get('unit') == 'GB'