diff --git a/scripts/uninstall_module.py b/scripts/uninstall_module.py index 439fca18a..8a9b70892 100755 --- a/scripts/uninstall_module.py +++ b/scripts/uninstall_module.py @@ -38,8 +38,9 @@ def uninstall_module(package_name: str, prefix=None): if prefix: add_site_dir(prefix) try: - dist_info_path = str(importlib_metadata.distribution(package_name)._path) - except importlib_metadata.PackageNotFoundError: + distribution = next(importlib_metadata.Distribution.discover(name=package_name)) + dist_info_path = str(distribution._path) + except StopIteration: print(f"Skipping {package_name} as it is not installed.") sys.exit(0)