diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/roles_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/roles_test.py b/tests/roles_test.py index a0cc26a..ad51995 100644 --- a/tests/roles_test.py +++ b/tests/roles_test.py @@ -25,4 +25,4 @@ def test_role_chat_only(): def test_explicit_image_roles(): role_names = load_ai_role_names('image') - assert set(role_names) == { 'hd-image' } + assert set(role_names) == { 'hd-image', 'hd', 'natural' } |