diff --git a/scripts/cutcrossentropy_install.py b/scripts/cutcrossentropy_install.py index b95875c00..396f4a655 100644 --- a/scripts/cutcrossentropy_install.py +++ b/scripts/cutcrossentropy_install.py @@ -18,12 +18,12 @@ if v < V("2.4.0"): cce_spec = importlib.util.find_spec("cut_cross_entropy") -uninstall_prefix = "" +UNINSTALL_PREFIX = "" if cce_spec: if not importlib.util.find_spec("cut_cross_entropy.transformers"): - uninstall_prefix = "pip uninstall -y cut-cross-entropy && " + UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && " print( - uninstall_prefix + UNINSTALL_PREFIX + 'pip install "cut-cross-entropy[transformers] @ git+https://github.com/apple/ml-cross-entropy.git@24fbe4b5dab9a6c250a014573613c1890190536c"' )