wip

2025-08-29 21:43:57 -04:00
parent 36c4ab11f9
commit 43f6f84269
2 changed files with 24 additions and 32 deletions
--- a/scripts/cutcrossentropy_install.py
+++ b/scripts/cutcrossentropy_install.py
@@ -23,7 +23,8 @@ cce_spec = importlib.util.find_spec("cut_cross_entropy")
 UNINSTALL_PREFIX = ""
 if cce_spec:
    if not importlib.util.find_spec("cut_cross_entropy.transformers"):
-        UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && "
+        uninstall_cmd = "uv pip uninstall" if USE_UV else "pip"
+        UNINSTALL_PREFIX = f"{uninstall_cmd} uninstall -y cut-cross-entropy && "

 UV_PREFIX = "uv " if USE_UV else ""