[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/test_model.py

@@ -33,8 +33,6 @@
 from transformers.models.qwen3 import Qwen3Config, Qwen3ForCausalLM
 from transformers.models.qwen3_moe import Qwen3MoeConfig, Qwen3MoeForCausalLM
 
-import litgpt.attention
-import litgpt.attention_utils
 import litgpt.config as config_module
 from litgpt import GPT, Config
 from litgpt.model import CausalSelfAttention
@@ -1470,7 +1468,8 @@ def assert_sdpa_backend(original_fn, query, k_and_v, mask, return_scores):
         pytest.xfail()
 
     model.mha.scaled_dot_product_attention = partial(
-        assert_sdpa_backend, model.mha.scaled_dot_product_attention,
+        assert_sdpa_backend,
+        model.mha.scaled_dot_product_attention,
     )
 
     if SUPPORTS_FLASH_ATTENTION:
@@ -1522,7 +1521,8 @@ def assert_sdpa_backend(original_fn, query, k_and_v, mask, return_scores):
         pytest.xfail()
 
     model.mha.scaled_dot_product_attention = partial(
-        assert_sdpa_backend, model.mha.scaled_dot_product_attention,
+        assert_sdpa_backend,
+        model.mha.scaled_dot_product_attention,
     )
 
     if SUPPORTS_FLASH_ATTENTION: