Lint

Author: lucbruni-amd

bitsandbytes/cextension.py

@@ -141,7 +141,9 @@ def __init__(self, error_msg: str):
         self.error_msg = error_msg
         self.user_cuda_version = get_cuda_version_tuple()
         self.available_versions = get_available_cuda_binary_versions()
-        self.override_value = os.environ.get("BNB_ROCM_VERSION") if HIP_ENVIRONMENT else os.environ.get("BNB_CUDA_VERSION")
+        self.override_value = (
+            os.environ.get("BNB_ROCM_VERSION") if HIP_ENVIRONMENT else os.environ.get("BNB_CUDA_VERSION")
+        )
         self.requested_version = (
             parse_cuda_version(self.override_value)
             if self.override_value

tests/test_cuda_setup_evaluator.py

@@ -58,7 +58,7 @@ def test_get_rocm_bnb_library_path_rejects_cuda_override(monkeypatch, rocm70_spe
     """BNB_CUDA_VERSION should be rejected on ROCm with a helpful error."""
     monkeypatch.delenv("BNB_ROCM_VERSION", raising=False)
     monkeypatch.setenv("BNB_CUDA_VERSION", "72")
-    with pytest.raises(RuntimeError, match="BNB_CUDA_VERSION.*detected for ROCm"):
+    with pytest.raises(RuntimeError, match=r"BNB_CUDA_VERSION.*detected for ROCm"):
         get_cuda_bnb_library_path(rocm70_spec)