print out the value of WITH_CUDA in the error

torchmd · Feb 27, 2025 · 2e83c31 · 2e83c31
1 parent fdd86d6
commit 2e83c31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 if env_with_cuda is not None:
     if env_with_cuda not in ("0", "1"):
         raise ValueError(
-            "Invalid flag with WITH_CUDA environment variable. Expected '0' or '1'"
+            f"WITH_CUDA environment variable got invalid value {env_with_cuda}. Expected '0' or '1'"
         )
     use_cuda = env_with_cuda == "1"
 else: