diff --git a/modules/extensions.py b/modules/extensions.py index 9a3e1f7f6..1063e5eb7 100644 --- a/modules/extensions.py +++ b/modules/extensions.py @@ -1,3 +1,4 @@ +from __future__ import annotations import os from datetime import datetime import git @@ -5,7 +6,7 @@ from modules import shared, errors from modules.paths import extensions_dir, extensions_builtin_dir -extensions = [] +extensions: list[Extension] = [] if not os.path.exists(extensions_dir): os.makedirs(extensions_dir) diff --git a/modules/ui_extensions.py b/modules/ui_extensions.py index 836fe5db6..a44211d7b 100644 --- a/modules/ui_extensions.py +++ b/modules/ui_extensions.py @@ -29,8 +29,8 @@ sort_ordering = { } -def get_installed(ext) -> extensions.Extension: - installed: extensions.Extension = [e for e in extensions.extensions if (e.remote or '').startswith(ext['url'].replace('.git', ''))] +def get_installed(ext): + installed = [e for e in extensions.extensions if (e.remote or '').startswith(ext['url'].replace('.git', ''))] return installed[0] if len(installed) > 0 else None