Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix static type errors with SCHEDULER_NAME_VALUES. And, avoid bi-directional cross-directory imports, which contribute to circular import issues.

Browse Source
This commit is contained in:
Ryan Dick
2024-07-03 11:13:16 -04:00
committed by Kent Keirsey
parent 3a24d70279
commit 35f8781ea2
8 changed files with 49 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/backend/stable_diffusion/schedulers/test_schedulers.py Normal file
View File

@ -0,0 +1,10 @@
from typing import get_args
from invokeai.backend.stable_diffusion.schedulers.schedulers import SCHEDULER_MAP, SCHEDULER_NAME_VALUES
def test_scheduler_map_has_all_keys():
# Assert that SCHEDULER_MAP has all keys from SCHEDULER_NAME_VALUES.
# TODO(ryand): This feels like it should be a type check, but I couldn't find a clean way to do this and didn't want
# to spend more time on it.
assert set(SCHEDULER_MAP.keys()) == set(get_args(SCHEDULER_NAME_VALUES))