make linter happy

Author: leofang

cuda_core/cuda/core/experimental/_module.py

@@ -245,7 +245,7 @@ class ObjectCode:
     __slots__ = ("_handle", "_backend_version", "_code_type", "_module", "_loader", "_sym_map")
     _supported_code_type = ("cubin", "ptx", "ltoir", "fatbin")
 
-    def __init__(self, module: Union[bytes, str], *, symbol_mapping: Optional[dict]=None):
+    def __init__(self, module: Union[bytes, str], *, symbol_mapping: Optional[dict] = None):
         _lazy_init()
 
         # handle is assigned during _lazy_load
@@ -256,7 +256,7 @@ def __init__(self, module: Union[bytes, str], *, symbol_mapping: Optional[dict]=
         self._module = module
         self._sym_map = {} if symbol_mapping is None else symbol_mapping
 
-    def _init(module, code_type, *, symbol_mapping: Optional[dict]=None):
+    def _init(module, code_type, *, symbol_mapping: Optional[dict] = None):
         self = ObjectCode.__new__(ObjectCode)
         if code_type not in self._supported_code_type:
             raise ValueError

cuda_core/tests/test_module.py

@@ -90,7 +90,7 @@ def test_object_code_load_cubin(get_saxpy_kernel):
     sym_map = mod._sym_map
     assert isinstance(cubin, bytes)
     mod = ObjectCode(cubin, symbol_mapping=sym_map)
-    ker = mod.get_kernel("saxpy<double>")
+    mod.get_kernel("saxpy<double>")  # force loading
 
 
 def test_object_code_load_cubin_from_file(get_saxpy_kernel, tmp_path):
@@ -101,4 +101,4 @@ def test_object_code_load_cubin_from_file(get_saxpy_kernel, tmp_path):
     cubin_file = tmp_path / "test.cubin"
     cubin_file.write_bytes(cubin)
     mod = ObjectCode(str(cubin_file), symbol_mapping=sym_map)
-    ker = mod.get_kernel("saxpy<double>")
+    mod.get_kernel("saxpy<double>")  # force loading