ViperEkura/AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: satisfy ruff format and import lint in setup.py

Browse Source 
- Merge nested if for CUDA version mismatch check
- Convert try-except-pass to return None (S110)
- Apply ruff format
This commit is contained in:
ViperEkura
2026-08-04 21:32:33 +08:00
parent 49a9c6b3d2
commit 6dffb0305a
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup.py
+6 -5
View File
@@ -62,8 +62,11 @@ class _CMakeBuildExt(_build_ext):
nvcc_ver = _cuda_toolkit_version() nvcc_ver = _cuda_toolkit_version()
torch_cuda = _torch_cuda_version() torch_cuda = _torch_cuda_version()
if nvcc_ver is not None and torch_cuda is not None: if (
if nvcc_ver[0] != int(torch_cuda.split(".")[0]): nvcc_ver is not None
and torch_cuda is not None
and nvcc_ver[0] != int(torch_cuda.split(".")[0])
):
warnings.warn( warnings.warn(
f"CUDA version mismatch: nvcc is {nvcc_ver[0]}.{nvcc_ver[1]} " f"CUDA version mismatch: nvcc is {nvcc_ver[0]}.{nvcc_ver[1]} "
f"but torch was built with CUDA {torch_cuda}. " f"but torch was built with CUDA {torch_cuda}. "
@@ -92,9 +95,7 @@ class _CMakeBuildExt(_build_ext):
if arch: if arch:
cfg.append(f"-DASTRAI_CUDA_ARCH={arch}") cfg.append(f"-DASTRAI_CUDA_ARCH={arch}")
subprocess.run(cfg, check=True) subprocess.run(cfg, check=True)
subprocess.run( subprocess.run([cmake, "--build", str(build_dir), "-j", parallel], check=True)
[cmake, "--build", str(build_dir), "-j", parallel], check=True
)
def _cuda_toolkit_version(): def _cuda_toolkit_version():