Lint

rapidsai · Nov 12, 2024 · 9f90408 · 9f90408
1 parent 70440be
commit 9f90408
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/python/kvikio/kvikio/__init__.py b/python/kvikio/kvikio/__init__.py
@@ -12,10 +12,10 @@
     del libkvikio
 
 
+from kvikio._lib.defaults import CompatMode  # noqa: F401
 from kvikio._version import __git_commit__, __version__
 from kvikio.cufile import CuFile
 from kvikio.remote_file import RemoteFile, is_remote_file_available
-from kvikio._lib.defaults import CompatMode
 
 __all__ = [
     "__git_commit__",

diff --git a/python/kvikio/kvikio/_lib/defaults.pyx b/python/kvikio/kvikio/_lib/defaults.pyx
@@ -4,8 +4,8 @@
 # distutils: language = c++
 # cython: language_level=3
 
-from libcpp cimport bool
 from libc.stdint cimport uint8_t
+from libcpp cimport bool
 
 
 cdef extern from "<kvikio/defaults.hpp>" namespace "kvikio" nogil:

diff --git a/python/kvikio/kvikio/defaults.py b/python/kvikio/kvikio/defaults.py
@@ -7,7 +7,7 @@
 import kvikio._lib.defaults
 
 
-def compat_mode() -> kvikio._lib.defaults.CompatMode:
+def compat_mode() -> kvikio.CompatMode:
     """Check if KvikIO is running in compatibility mode.
 
     Notice, this is not the same as the compatibility mode in cuFile. That is,
@@ -31,7 +31,7 @@ def compat_mode() -> kvikio._lib.defaults.CompatMode:
     return kvikio._lib.defaults.compat_mode()
 
 
-def compat_mode_reset(compatmode: kvikio._lib.defaults.CompatMode) -> None:
+def compat_mode_reset(compatmode: kvikio.CompatMode) -> None:
     """Reset the compatibility mode.
 
     Use this function to enable/disable compatibility mode explicitly.
@@ -47,7 +47,7 @@ def compat_mode_reset(compatmode: kvikio._lib.defaults.CompatMode) -> None:
 
 
 @contextlib.contextmanager
-def set_compat_mode(compatmode: kvikio._lib.defaults.CompatMode):
+def set_compat_mode(compatmode: kvikio.CompatMode):
     """Context for resetting the compatibility mode.
 
     Parameters

diff --git a/python/kvikio/tests/test_defaults.py b/python/kvikio/tests/test_defaults.py
@@ -15,7 +15,7 @@ def test_compat_mode():
     """Test changing `compat_mode`"""
 
     before = kvikio.defaults.compat_mode()
-    with kvikio.defaults.set_compat_mode():
+    with kvikio.defaults.set_compat_mode(kvikio.CompatMode.ON):
         assert kvikio.defaults.compat_mode()
         kvikio.defaults.compat_mode_reset(kvikio.CompatMode.OFF)
         assert not kvikio.defaults.compat_mode()