From f4e33ff1c731265ebd5324b3670312d3f7b93ad7 Mon Sep 17 00:00:00 2001
From: Artificial Sweetener <artificialsweetenerai@proton.me>
Date: Sat, 5 Sep 2026 18:25:59 -0400
Subject: [PATCH] ci(lint): stabilize first-party import classification

---
 pyproject.toml                                 | 3 +++
 tests/test_model_attn1_replacement_mutation.py | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index b6fcfd7..945b94b 100644
--- a/pyproject.toml
+++ b/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
diff --git a/tests/test_model_attn1_replacement_mutation.py b/tests/test_model_attn1_replacement_mutation.py
index be5ab4b..5e71730 100644
--- a/tests/test_model_attn1_replacement_mutation.py
+++ b/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: