Unify ruff config and apply lint fixes across all sub-repos

ci/tools/merge_cuda_core_wheels.py

@@ -30,7 +30,7 @@
 from typing import List
 
 
-def run_command(cmd: List[str], cwd: Path = None, env: dict = os.environ) -> subprocess.CompletedProcess:
+def run_command(cmd: List[str], cwd: Path | None = None, env: dict = os.environ) -> subprocess.CompletedProcess:
     """Run a command with error handling."""
     print(f"Running: {' '.join(cmd)}")
     if cwd:
@@ -47,7 +47,7 @@ def run_command(cmd: List[str], cwd: Path = None, env: dict = os.environ) -> sub
     return result
 
 
-def print_wheel_directory_structure(wheel_path: Path, filter_prefix: str = "cuda/core/", label: str = None):
+def print_wheel_directory_structure(wheel_path: Path, filter_prefix: str = "cuda/core/", label: str | None = None):
     """Print the directory structure of a wheel file, similar to unzip -l output.
 
     Args:

conftest.py

@@ -6,7 +6,7 @@
 import pytest
 
 
-def pytest_collection_modifyitems(config, items):
+def pytest_collection_modifyitems(config, items):  # noqa: ARG001
     cuda_home = os.environ.get("CUDA_HOME")
     for item in items:
         nodeid = item.nodeid.replace("\\", "/")

cuda_bindings/benchmarks/conftest.py

@@ -3,6 +3,7 @@
 
 import numpy as np
 import pytest
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import nvrtc
 from cuda.bindings import runtime as cudart
@@ -22,7 +23,7 @@ def ASSERT_DRV(err):
         raise RuntimeError(f"Unknown error type: {err}")
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def init_cuda():
     # Initialize
     (err,) = cuda.cuInit(0)
@@ -44,7 +45,7 @@ def init_cuda():
     ASSERT_DRV(err)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def load_module():
     module = None
 

cuda_bindings/benchmarks/test_launch_latency.py

@@ -4,10 +4,10 @@
 import ctypes
 
 import pytest
-from cuda.bindings import driver as cuda
 from kernels import kernel_string
 
 from conftest import ASSERT_DRV
+from cuda.bindings import driver as cuda
 
 
 def launch(kernel, stream, args=(), arg_types=()):

cuda_bindings/benchmarks/test_pointer_attributes.py

@@ -4,9 +4,9 @@
 import random
 
 import pytest
-from cuda.bindings import driver as cuda
 
 from conftest import ASSERT_DRV
+from cuda.bindings import driver as cuda
 
 random.seed(0)
 

cuda_bindings/build_hooks.py

@@ -176,20 +176,20 @@ def _parse_headers(header_dict, include_path_list, parser_caching):
             CUDA_VERSION = parser.defs["macros"].get("CUDA_VERSION", "Unknown")
             print(f"Found CUDA_VERSION: {CUDA_VERSION}", flush=True)
 
-        found_types += {key for key in parser.defs["types"]}
-        found_types += {key for key in parser.defs["structs"]}
-        found_types += {key for key in parser.defs["unions"]}
-        found_types += {key for key in parser.defs["enums"]}
-        found_functions += {key for key in parser.defs["functions"]}
-        found_values += {key for key in parser.defs["values"]}
+        found_types += set(parser.defs["types"])
+        found_types += set(parser.defs["structs"])
+        found_types += set(parser.defs["unions"])
+        found_types += set(parser.defs["enums"])
+        found_functions += set(parser.defs["functions"])
+        found_values += set(parser.defs["values"])
 
         for key, value in parser.defs["structs"].items():
             struct_list[key] = _Struct(key, value["members"])
         for key, value in parser.defs["unions"].items():
             struct_list[key] = _Struct(key, value["members"])
 
         for key, value in struct_list.items():
-            if key.startswith("anon_union") or key.startswith("anon_struct"):
+            if key.startswith(("anon_union", "anon_struct")):
                 continue
 
             found_struct += [key]
@@ -408,7 +408,7 @@ def _cleanup_dst_files():
         )
 
     # Cythonize
-    cython_directives = dict(language_level=3, embedsignature=True, binding=True, freethreading_compatible=True)
+    cython_directives = {"language_level": 3, "embedsignature": True, "binding": True, "freethreading_compatible": True}
     if compile_for_coverage:
         cython_directives["linetrace"] = True
 

cuda_bindings/cuda/bindings/_test_helpers/arch_check.py

@@ -6,6 +6,7 @@
 from functools import cache
 
 import pytest
+
 from cuda.bindings import nvml
 
 

cuda_bindings/examples/0_Introduction/clock_nvrtc_test.py

@@ -6,6 +6,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
+
 from cuda.bindings import driver as cuda
 
 clock_nvrtc = """\
@@ -67,7 +68,7 @@ def main():
     timer = np.empty(NUM_BLOCKS * 2, dtype="int64")
     hinput = np.empty(NUM_THREADS * 2, dtype="float32")
 
-    for i in range(0, NUM_THREADS * 2):
+    for i in range(NUM_THREADS * 2):
         hinput[i] = i
 
     devID = findCudaDevice()
@@ -105,7 +106,7 @@ def main():
 
     avgElapsedClocks = 0.0
 
-    for i in range(0, NUM_BLOCKS):
+    for i in range(NUM_BLOCKS):
         avgElapsedClocks += timer[i + NUM_BLOCKS] - timer[i]
 
     avgElapsedClocks = avgElapsedClocks / NUM_BLOCKS

cuda_bindings/examples/0_Introduction/simpleCubemapTexture_test.py

@@ -8,6 +8,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/0_Introduction/simpleP2P_test.py

@@ -8,6 +8,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/0_Introduction/simpleZeroCopy_test.py

@@ -11,6 +11,7 @@
 from common import common
 from common.helper_cuda import checkCudaErrors
 from common.helper_string import checkCmdLineFlag, getCmdLineArgumentInt
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/0_Introduction/systemWideAtomics_test.py

@@ -8,6 +8,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/0_Introduction/vectorAddDrv_test.py

@@ -8,6 +8,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDeviceDRV
+
 from cuda.bindings import driver as cuda
 
 vectorAddDrv = """\

cuda_bindings/examples/0_Introduction/vectorAddMMAP_test.py

@@ -9,6 +9,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDeviceDRV
+
 from cuda.bindings import driver as cuda
 
 vectorAddMMAP = """\

cuda_bindings/examples/2_Concepts_and_Techniques/streamOrderedAllocation_test.py

@@ -11,6 +11,7 @@
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
 from common.helper_string import checkCmdLineFlag
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/3_CUDA_Features/globalToShmemAsyncCopy_test.py

@@ -11,6 +11,7 @@
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
 from common.helper_string import checkCmdLineFlag, getCmdLineArgumentInt
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/3_CUDA_Features/simpleCudaGraphs_test.py

@@ -7,6 +7,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 
@@ -116,7 +117,7 @@
 def init_input(a, size):
     ctypes.c_float.from_address(a)
     a_list = ctypes.pointer(ctypes.c_float.from_address(a))
-    for i in range(0, size):
+    for i in range(size):
         a_list[i] = rnd.random()
 
 

cuda_bindings/examples/4_CUDA_Libraries/conjugateGradientMultiBlockCG_test.py

@@ -10,6 +10,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors, findCudaDevice
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/common/common.py

@@ -4,6 +4,7 @@
 
 import numpy as np
 from common.helper_cuda import checkCudaErrors
+
 from cuda import pathfinder
 from cuda.bindings import driver as cuda
 from cuda.bindings import nvrtc

cuda_bindings/examples/common/helper_cuda.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
 from common.helper_string import checkCmdLineFlag, getCmdLineArgumentInt
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import nvrtc
 from cuda.bindings import runtime as cudart

cuda_bindings/examples/extra/isoFDModelling_test.py

@@ -6,6 +6,7 @@
 import numpy as np
 from common import common
 from common.helper_cuda import checkCudaErrors
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import runtime as cudart
 

cuda_bindings/examples/extra/jit_program_test.py

@@ -4,6 +4,7 @@
 import ctypes
 
 import numpy as np
+
 from cuda.bindings import driver as cuda
 from cuda.bindings import nvrtc
 

cuda_bindings/examples/extra/numba_emm_plugin.py

@@ -49,15 +49,16 @@
 
 from ctypes import c_size_t
 
-from cuda.bindings import driver as cuda
-from cuda.bindings import driver as cuda_driver
 from numba.cuda import (
     GetIpcHandleMixin,
     HostOnlyCUDAMemoryManager,
     MemoryInfo,
     MemoryPointer,
 )
 
+from cuda.bindings import driver as cuda
+from cuda.bindings import driver as cuda_driver
+
 # Python functions for allocation, deallocation, and memory info via the NVIDIA
 # CUDA Python Driver API
 

cuda_bindings/tests/conftest.py

@@ -5,9 +5,10 @@
 import sys
 from importlib.metadata import PackageNotFoundError, distribution
 
-import cuda.bindings.driver as cuda
 import pytest
 
+import cuda.bindings.driver as cuda
+
 # Import shared test helpers for tests across subprojects.
 # PLEASE KEEP IN SYNC with copies in other conftest.py in this repo.
 _test_helpers_root = pathlib.Path(__file__).resolve().parents[2] / "cuda_python_test_helpers"

cuda_bindings/tests/nvml/__init__.py

@@ -3,6 +3,7 @@
 
 
 import pytest
+
 from cuda.bindings._test_helpers.arch_check import hardware_supports_nvml
 
 if not hardware_supports_nvml():

cuda_bindings/tests/nvml/conftest.py

@@ -4,6 +4,7 @@
 from collections import namedtuple
 
 import pytest
+
 from cuda.bindings import nvml
 from cuda.bindings._test_helpers.arch_check import unsupported_before  # noqa: F401
 
@@ -74,7 +75,7 @@ def get_devices(device_info):
 @pytest.fixture
 def all_devices(device_info):
     with NVMLInitializer():
-        yield sorted(list(set(get_devices(device_info))))
+        yield sorted(set(get_devices(device_info)))
 
 
 @pytest.fixture

cuda_bindings/tests/nvml/test_compute_mode.py

@@ -5,6 +5,7 @@
 import sys
 
 import pytest
+
 from cuda.bindings import nvml
 
 from .conftest import unsupported_before

cuda_bindings/tests/nvml/test_device.py

@@ -5,6 +5,7 @@
 from functools import cache
 
 import pytest
+
 from cuda.bindings import nvml
 
 from .conftest import unsupported_before

cuda_bindings/tests/nvml/test_gpu.py

@@ -3,6 +3,7 @@
 
 import numpy as np
 import pytest
+
 from cuda.bindings import nvml
 
 from . import util

cuda_bindings/tests/nvml/test_init.py

@@ -5,6 +5,7 @@
 import warnings
 
 import pytest
+
 from cuda.bindings import nvml
 
 
@@ -24,10 +25,10 @@ def test_devices_are_the_same_architecture(all_devices):
     # they won't be tested properly.  This tests for the (hopefully rare) case
     # where a system has devices of different architectures and produces a warning.
 
-    all_arches = set(nvml.DeviceArch(nvml.device_get_architecture(device)) for device in all_devices)
+    all_arches = {nvml.DeviceArch(nvml.device_get_architecture(device)) for device in all_devices}
 
     if len(all_arches) > 1:
-        warnings.warn(  # noqa: B028
+        warnings.warn(
             f"System has devices of multiple architectures ({', '.join(x.name for x in all_arches)}). "
             f" Some tests may be skipped unexpectedly",
             UserWarning,

cuda_bindings/tests/nvml/test_page_retirement.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
 import pytest
+
 from cuda.bindings import nvml
 
 from . import util

cuda_bindings/tests/nvml/test_pynvml.py

@@ -7,6 +7,7 @@
 import time
 
 import pytest
+
 from cuda.bindings import nvml
 
 from . import util

cuda_bindings/tests/test_basics.py

@@ -5,6 +5,7 @@
 import sys
 
 import pytest
+
 from cuda.bindings._internal import _fast_enum
 
 # Test both with the FastEnum implementation and the stdlib enum.IntEnum (even