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