Privatize pathfinder driver info API in compatibility branch

Align the compatibility-checks branch with the internal-only driver info helper by dropping the public re-exports and updating docs and tests to use the internal module directly.

Made-with: Cursor

Author: rwgk

cuda_pathfinder/cuda/pathfinder/__init__.py

@@ -68,9 +68,6 @@
 from cuda.pathfinder._static_libs.find_static_lib import (
     locate_static_lib as locate_static_lib,
 )
-from cuda.pathfinder._utils.driver_info import DriverCudaVersion as DriverCudaVersion
-from cuda.pathfinder._utils.driver_info import QueryDriverCudaVersionError as QueryDriverCudaVersionError
-from cuda.pathfinder._utils.driver_info import query_driver_cuda_version as query_driver_cuda_version
 from cuda.pathfinder._utils.env_vars import get_cuda_path_or_home as get_cuda_path_or_home
 
 from cuda.pathfinder._version import __version__  # isort: skip

cuda_pathfinder/docs/source/api.rst

@@ -22,10 +22,6 @@ CUDA bitcode and static libraries.
    CompatibilityCheckError
    CompatibilityInsufficientMetadataError
 
-   DriverCudaVersion
-   QueryDriverCudaVersionError
-   query_driver_cuda_version
-
    SUPPORTED_NVIDIA_LIBNAMES
    load_nvidia_dynamic_lib
    LoadedDL

cuda_pathfinder/tests/test_utils_driver_info.py

@@ -5,15 +5,6 @@
 
 import pytest
 
-from cuda.pathfinder import (
-    DriverCudaVersion as PublicDriverCudaVersion,
-)
-from cuda.pathfinder import (
-    QueryDriverCudaVersionError as PublicQueryDriverCudaVersionError,
-)
-from cuda.pathfinder import (
-    query_driver_cuda_version as public_query_driver_cuda_version,
-)
 from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL
 from cuda.pathfinder._utils import driver_info
 
@@ -51,12 +42,6 @@ def _loaded_cuda(abs_path: str) -> LoadedDL:
     )
 
 
-def test_driver_cuda_version_public_api_exports():
-    assert PublicDriverCudaVersion is driver_info.DriverCudaVersion
-    assert PublicQueryDriverCudaVersionError is driver_info.QueryDriverCudaVersionError
-    assert public_query_driver_cuda_version is driver_info.query_driver_cuda_version
-
-
 def test_query_driver_cuda_version_uses_windll_on_windows(monkeypatch):
     fake_driver_lib = _FakeDriverLib(status=0, version=12080)
     loaded_paths: list[str] = []

cuda_pathfinder/tests/test_with_compatibility_checks.py

@@ -12,16 +12,15 @@
     BitcodeLibNotFoundError,
     CompatibilityCheckError,
     CompatibilityInsufficientMetadataError,
-    DriverCudaVersion,
     DynamicLibNotFoundError,
     LoadedDL,
     LocatedBitcodeLib,
     LocatedHeaderDir,
     LocatedStaticLib,
-    QueryDriverCudaVersionError,
     StaticLibNotFoundError,
     WithCompatibilityChecks,
 )
+from cuda.pathfinder._utils.driver_info import DriverCudaVersion, QueryDriverCudaVersionError
 
 STRICTNESS = os.environ.get("CUDA_PATHFINDER_TEST_WITH_COMPATIBILITY_CHECKS_STRICTNESS", "see_what_works")
 assert STRICTNESS in ("see_what_works", "all_must_work")