Use XOR when checking NVML init flags

dask · Apr 18, 2022 · 5c4e74a · 5c4e74a
1 parent 15c483e
commit 5c4e74a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/distributed/diagnostics/tests/test_nvml.py b/distributed/diagnostics/tests/test_nvml.py
@@ -39,9 +39,9 @@ def test_enable_disable_nvml():
     with dask.config.set({"distributed.diagnostics.nvml": True}):
         nvml.init_once()
         assert (
-            nvml.nvmlInitialized is True
-            or nvml.nvmlLibraryNotFound is True
-            or nvml.nvmlWslInsufficientDriver is True
+            nvml.nvmlInitialized
+            ^ nvml.nvmlLibraryNotFound
+            ^ nvml.nvmlWslInsufficientDriver
         )