style: fix ruff lint warnings

- Remove unused local variable b in attention_backend.py - Remove unused variable rank0_sd in test_broadcast_state_dict.py - Remove unused imports across test files
2026-08-07 14:17:48 +08:00
parent ef1bb6f401
commit 55ee258e95
6 changed files with 1 additions and 10 deletions
@@ -10,7 +10,6 @@ from astrai.extension import (
    ATTN_BACKEND,
    AttentionBackendFactory,
    CudaBackend,
-    TorchNativeBackend,
    attn_backend,
    get_backend,
 )