Skip to content

Commit f039cfe

Browse files
committed
fix format
1 parent 95387c8 commit f039cfe

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

bitsandbytes/__init__.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
matmul_cublas,
1818
mm_cublas,
1919
)
20-
from .backends import register_backend
20+
from .backends import backends, register_backend
2121
from .backends.cpu import CPUBackend
2222
from .backends.npu import NPUBackend
2323
from .cextension import lib
@@ -65,7 +65,8 @@
6565

6666

6767
# import module after decided backends
68-
from .nn import modules
68+
if backends:
69+
from .nn import modules
6970

7071
# TODO: Other potential backends:
7172
# XLA - Google TPU / PJRT runtime

setup.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,8 @@ def write_version_file(version, filepath="bitsandbytes/_version.py"):
3737

3838

3939
def get_version_and_write_to_file():
40-
try:
41-
latest_semver_tag = get_latest_semver_tag()
42-
version = latest_semver_tag if is_git_tagged_commit() else f"{latest_semver_tag}.dev+{get_git_commit_hash()}"
43-
except:
44-
print("Cannot get version by git tag, use 1.0 defaultly")
45-
version = "1.0.dev+0"
40+
latest_semver_tag = get_latest_semver_tag()
41+
version = latest_semver_tag if is_git_tagged_commit() else f"{latest_semver_tag}.dev+{get_git_commit_hash()}"
4642
write_version_file(version)
4743
return version
4844

0 commit comments

Comments
 (0)