ci(lint): stabilize first-party import classification

pyproject.toml

@@ -43,6 +43,9 @@ extend-exclude = [
 select = ["E", "F", "I", "UP", "B", "C4", "ANN"]
 ignore = ["ANN401"]
 
+[tool.ruff.lint.isort]
+known-first-party = ["simple_syrup"]
+
 [tool.mypy]
 python_version = "3.12"
 warn_return_any = true

tests/test_model_attn1_replacement_mutation.py

@@ -11,10 +11,11 @@ from typing import Any
 import pytest
 import torch
 from comfy.model_patcher import ModelPatcher
+from torch import nn
+
 from simple_syrup.runtime.model_attention_replacement_mutations import (
     ModelAttn1ReplacementsMutation,
 )
-from torch import nn
 
 
 def test_attn1_replacements_install_every_declared_key_atomically() -> None: