xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 03:50:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: mock torch in cache middleware tests

Browse Source
This commit is contained in:
Arjan Singh 2025-08-28 18:06:41 -07:00
parent 6c26a501c8
commit 180e19e923
No known key found for this signature in database
GPG Key ID: B1102A5F9699979D
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests-unit/server_test/test_cache_control.py
View File

@ -12,6 +12,8 @@ with (
patch("app.frontend_management.FrontendManager"),
patch("utils.install_util.get_missing_requirements_message"),
patch("utils.install_util.requirements_path"),
patch("comfy.model_management.get_torch_device"),
patch("comfy.model_management.get_total_memory", return_value=1024 * 1024 * 1024),
):
from server import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS