aboutsummaryrefslogtreecommitdiff
path: root/modules/launch_utils.py
diff options
context:
space:
mode:
authorKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-04 16:47:41 +0800
committerKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-04 16:47:41 +0800
commitf5f89780ccf637979c0c88546080a13afd61c9a3 (patch)
tree4ba683c9d8884193960b1dbe74809bf6c77e23c2 /modules/launch_utils.py
parent9a15ae2a92e55d614fe515cd0a104d90b854b23f (diff)
parent48fae7ccdc2fe2d2ba8e8cfcb17b56028734e570 (diff)
Merge branch 'dev' into test-fp8
Diffstat (limited to 'modules/launch_utils.py')
-rw-r--r--modules/launch_utils.py14
1 files changed, 10 insertions, 4 deletions
diff --git a/modules/launch_utils.py b/modules/launch_utils.py
index 031488a0..2c54e2a0 100644
--- a/modules/launch_utils.py
+++ b/modules/launch_utils.py
@@ -6,6 +6,7 @@ import os
import shutil
import sys
import importlib.util
+import importlib.metadata
import platform
import json
from functools import lru_cache
@@ -119,11 +120,16 @@ def run(command, desc=None, errdesc=None, custom_env=None, live: bool = default_
def is_installed(package):
try:
- spec = importlib.util.find_spec(package)
- except ModuleNotFoundError:
- return False
+ dist = importlib.metadata.distribution(package)
+ except importlib.metadata.PackageNotFoundError:
+ try:
+ spec = importlib.util.find_spec(package)
+ except ModuleNotFoundError:
+ return False
+
+ return spec is not None
- return spec is not None
+ return dist is not None
def repo_dir(name):