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
@@ -5,7 +5,6 @@ import torch

 from astrai.config.model_config import AutoRegressiveLMConfig
 from astrai.model.transformer import AutoRegressiveLM
-from tests.conftest import skip_no_kernel

 D = 64
 CFG = dict(