pre-commit run --all-files

Author: Titus-von-Koeller

bitsandbytes/__init__.py

@@ -4,7 +4,6 @@
 # LICENSE file in the root directory of this source tree.
 
 from . import research, utils
-from .cextension import lib
 from .autograd._functions import (
     MatmulLtState,
     bmm_cublas,
@@ -13,12 +12,14 @@
     matmul_cublas,
     mm_cublas,
 )
+from .cextension import lib
 from .nn import modules
 
 if lib and lib.compiled_with_cuda:
     from .backends import register_backend
     from .backends.cuda import CUDABackend
     from .optim import adam
+
     register_backend("cuda", CUDABackend())
 __pdoc__ = {
     "libbitsandbytes": False,

bitsandbytes/backends/__init__.py

@@ -4,9 +4,11 @@
 
 backends: Dict[str, Backend] = {}
 
+
 def register_backend(backend_name: str, backend_instance: Backend):
     backends[backend_name.lower()] = backend_instance
 
+
 def ensure_backend_is_available(device_type: str):
     """Check if a backend is available for the given device type."""
     if device_type.lower() not in backends: