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(db): raise a MigrationVersionError when invalid versions are used

Browse Source 
This inherits from `ValueError`, so pydantic understands it when doing validation.
This commit is contained in:
psychedelicious 2023-12-13 11:21:16 +11:00
parent ebf5f5d418
commit c50a49719b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/services/shared/sqlite_migrator/sqlite_migrator_common.py
View File

@ -97,7 +97,7 @@ class Migration(BaseModel):
def validate_to_version(self) -> "Migration":
"""Validates that to_version is one greater than from_version."""
if self.to_version != self.from_version + 1:
raise ValueError("to_version must be one greater than from_version")
raise MigrationVersionError("to_version must be one greater than from_version")
return self
def __hash__(self) -> int: