Skip to content

Commit 583af91

Browse files
committed
pre-commit fixes
1 parent 8900cd2 commit 583af91

1 file changed

Lines changed: 8 additions & 10 deletions

File tree

cuda_pathfinder/tests/test_compatibility_guard_rails.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,11 @@
66
from pathlib import Path
77

88
import pytest
9+
from local_helpers import (
10+
locate_real_cuda_toolkit_version_from_cuda_h,
11+
require_real_cuda_toolkit_version_from_cuda_h,
12+
require_real_driver_cuda_version,
13+
)
914

1015
import cuda.pathfinder._compatibility_guard_rails as compatibility_module
1116
from cuda import pathfinder
@@ -26,15 +31,10 @@
2631
from cuda.pathfinder._headers.find_nvidia_headers import (
2732
locate_nvidia_header_directory as locate_nvidia_header_directory_raw,
2833
)
29-
from cuda.pathfinder._utils.env_vars import get_cuda_path_or_home
3034
from cuda.pathfinder._utils import driver_info
3135
from cuda.pathfinder._utils.driver_info import DriverCudaVersion, QueryDriverCudaVersionError
36+
from cuda.pathfinder._utils.env_vars import get_cuda_path_or_home
3237
from cuda.pathfinder._utils.toolkit_info import read_cuda_header_version
33-
from local_helpers import (
34-
locate_real_cuda_toolkit_version_from_cuda_h,
35-
require_real_driver_cuda_version,
36-
require_real_cuda_toolkit_version_from_cuda_h,
37-
)
3838

3939
STRICTNESS = os.environ.get("CUDA_PATHFINDER_TEST_COMPATIBILITY_GUARD_RAILS_STRICTNESS", "see_what_works")
4040
assert STRICTNESS in ("see_what_works", "all_must_work")
@@ -693,8 +693,7 @@ def test_real_wheel_ctk_items_are_compatible(info_summary_append):
693693
f"via {real_ctk.found_via} at {real_ctk.cuda_h_path!r}"
694694
)
695695
info_summary_append(
696-
"real driver CUDA version="
697-
f"{real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
696+
f"real driver CUDA version={real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
698697
)
699698
guard_rails = CompatibilityGuardRails(
700699
ctk_major=real_ctk.version.major,
@@ -742,8 +741,7 @@ def test_real_wheel_component_version_does_not_override_ctk_line(info_summary_ap
742741
f"via {real_ctk.found_via} at {real_ctk.cuda_h_path!r}"
743742
)
744743
info_summary_append(
745-
"real driver CUDA version="
746-
f"{real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
744+
f"real driver CUDA version={real_driver.major}.{real_driver.minor} (encoded={real_driver.encoded})"
747745
)
748746
guard_rails = CompatibilityGuardRails(
749747
ctk_major=real_ctk.version.major,

0 commit comments

Comments
 (0)