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(config): fix config _check_for_discontinuities

Browse Source 
Need to sort the migrations first.
This commit is contained in:
psychedelicious 2024-05-14 16:55:22 +10:00
parent 4c081d58e0
commit d487102904
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/services/config/config_migrate.py
View File

@ -9,6 +9,7 @@ import shutil
from copy import deepcopy from copy import deepcopy
from functools import lru_cache from functools import lru_cache
from pathlib import Path from pathlib import Path
from typing import Iterable
import yaml import yaml
from packaging.version import Version from packaging.version import Version
@ -37,9 +38,10 @@ class ConfigMigrator:
self._migrations.add(migration) self._migrations.add(migration)
@staticmethod @staticmethod
def _check_for_discontinuities(migrations: list[ConfigMigration]) -> None: def _check_for_discontinuities(migrations: Iterable[ConfigMigration]) -> None:
current_version = Version("3.0.0") current_version = Version("3.0.0")
for m in migrations: sorted_migrations = sorted(migrations, key=lambda x: x.from_version)
for m in sorted_migrations:
if current_version != m.from_version: if current_version != m.from_version:
raise ValueError( raise ValueError(
f"Migration functions are not continuous. Expected from_version={current_version} but got from_version={m.from_version}, for migration function {m.function.__name__}" f"Migration functions are not continuous. Expected from_version={current_version} but got from_version={m.from_version}, for migration function {m.function.__name__}"