summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_cuda_setup_evaluator.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/test_cuda_setup_evaluator.py b/tests/test_cuda_setup_evaluator.py
index 3d34c29..c947ca1 100644
--- a/tests/test_cuda_setup_evaluator.py
+++ b/tests/test_cuda_setup_evaluator.py
@@ -133,7 +133,7 @@ def test_full_system():
)
version = float(f"{major}.{minor}")
- if version == "" and "LD_LIBRARY_PATH":
+ if version == "" and "LD_LIBRARY_PATH" in os.environ:
ld_path = os.environ["LD_LIBRARY_PATH"]
paths = ld_path.split(":")
version = ""
@@ -144,6 +144,7 @@ def test_full_system():
version = float(version)
break
+
assert version > 0
binary_name = evaluate_cuda_setup()
binary_name = binary_name.replace("libbitsandbytes_cuda", "")