diff --git a/utils/__init__.py b/utils/__init__.py index 9bc06335de..a855c361d1 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -25,7 +25,7 @@ def get_pkg_versions(packages: List[str]) -> Dict[str, str]: versions = {} for module in packages: cmd = [sys.executable, "-c", f"import {module}; print({module}.__version__)"] - version = subprocess.check_output(cmd).decode().strip() + version = subprocess.check_output(cmd).decode().split("+")[0].strip() versions[module] = version return versions