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

chore: ruff lint

Browse Source
This commit is contained in:
psychedelicious
2023-11-14 07:57:07 +11:00
parent 4465f97cdf
commit bc64cde6f9
3 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/app/services/model_records/test_model_records_sql.py
View File

@ -52,16 +52,16 @@ def test_type(store: ModelRecordServiceBase):
def test_add(store: ModelRecordServiceBase):
raw = dict(
path="/tmp/foo.ckpt",
name="model1",
base=BaseModelType("sd-1"),
type="main",
config="/tmp/foo.yaml",
variant="normal",
format="checkpoint",
original_hash="111222333444",
)
raw = {
"path": "/tmp/foo.ckpt",
"name": "model1",
"base": BaseModelType("sd-1"),
"type": "main",
"config": "/tmp/foo.yaml",
"variant": "normal",
"format": "checkpoint",
"original_hash": "111222333444",
}
store.add_model("key1", raw)
config1 = store.get_model("key1")
assert config1 is not None