From 1ac6aaf6a6969865e23bb5c6150f3b58f7fcd7b0 Mon Sep 17 00:00:00 2001
From: "H. Jhoo" <hoyoung.jhoo@sf.snu.ac.kr>
Date: Wed, 7 May 2025 21:30:58 +0900
Subject: [PATCH] Remove + signs from package versions

From the latest PyTorch installation via uv/pip, `torch.__version__` contains a cuda version appendix such as  `2.7.0+cu126`. This occurs a pip dependency conflict.
---
 utils/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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