Fix ruff formatting and import sorting for CI lint pass

- Auto-format 7 files with ruff format (test files + agentic modules)
- Fix import sorting in test_models.py (ruff I001)

https://claude.ai/code/session_01KhAhJKpEoqCzmWzcALSfW6

Author: claude

agentic/workflows/modules/agent.py

@@ -144,7 +144,7 @@ def parse_json(output: str, target_type: Type[T] = None) -> Any:
                 return [target_type.model_validate(item) for item in parsed]
             return target_type.model_validate(parsed)
         return parsed
-    except (json.JSONDecodeError, ValueError):
+    except json.JSONDecodeError, ValueError:
         pass
 
     # Strategy 2: Strip markdown code fences
@@ -162,7 +162,7 @@ def parse_json(output: str, target_type: Type[T] = None) -> Any:
                         return [target_type.model_validate(item) for item in parsed]
                     return target_type.model_validate(parsed)
                 return parsed
-            except (json.JSONDecodeError, ValueError):
+            except json.JSONDecodeError, ValueError:
                 pass
 
     # Strategy 3: Find first JSON array or object in output
@@ -182,7 +182,7 @@ def parse_json(output: str, target_type: Type[T] = None) -> Any:
                     return [target_type.model_validate(item) for item in parsed]
                 return target_type.model_validate(parsed)
             return parsed
-        except (json.JSONDecodeError, ValueError):
+        except json.JSONDecodeError, ValueError:
             continue
 
     raise json.JSONDecodeError("No valid JSON found in output", output, 0)

agentic/workflows/modules/orchestrator.py

@@ -45,5 +45,5 @@ def extract_run_id(stdout: str) -> str | None:
     try:
         data = json.loads(stdout.strip())
         return data.get("run_id")
-    except (json.JSONDecodeError, ValueError):
+    except json.JSONDecodeError, ValueError:
         return None

agentic/workflows/modules/state.py

@@ -172,7 +172,7 @@ def from_stdin(cls) -> Optional["WorkflowState"]:
             state = cls(run_id=run_id, prompt=data.get("prompt", ""))
             state.data = data
             return state
-        except (json.JSONDecodeError, EOFError):
+        except json.JSONDecodeError, EOFError:
             return None
 
     def to_stdout(self) -> None:

tests/unit/api/test_plots_helpers.py

@@ -134,19 +134,13 @@ def test_single_group_no_match(self) -> None:
     def test_multiple_groups_and_logic(self) -> None:
         spec_lookup = {"s1": {"tags": {"plot_type": ["scatter"], "domain": ["statistics"]}}}
         impl_lookup = {}
-        groups = [
-            {"category": "plot", "values": ["scatter"]},
-            {"category": "dom", "values": ["statistics"]},
-        ]
+        groups = [{"category": "plot", "values": ["scatter"]}, {"category": "dom", "values": ["statistics"]}]
         assert _image_matches_groups("s1", "matplotlib", groups, spec_lookup, impl_lookup) is True
 
     def test_multiple_groups_one_fails(self) -> None:
         spec_lookup = {"s1": {"tags": {"plot_type": ["scatter"], "domain": ["finance"]}}}
         impl_lookup = {}
-        groups = [
-            {"category": "plot", "values": ["scatter"]},
-            {"category": "dom", "values": ["statistics"]},
-        ]
+        groups = [{"category": "plot", "values": ["scatter"]}, {"category": "dom", "values": ["statistics"]}]
         assert _image_matches_groups("s1", "matplotlib", groups, spec_lookup, impl_lookup) is False
 
     def test_spec_not_in_lookup(self) -> None:
@@ -223,10 +217,7 @@ def test_single_group(self) -> None:
         assert result == "filter:lib=matplotlib"
 
     def test_multiple_groups_sorted(self) -> None:
-        groups = [
-            {"category": "plot", "values": ["scatter"]},
-            {"category": "lib", "values": ["matplotlib"]},
-        ]
+        groups = [{"category": "plot", "values": ["scatter"]}, {"category": "lib", "values": ["matplotlib"]}]
         result = _build_cache_key(groups)
         assert result == "filter:lib=matplotlib:plot=scatter"
 
@@ -236,14 +227,8 @@ def test_values_sorted(self) -> None:
         assert result == "filter:lib=matplotlib,seaborn"
 
     def test_stable_key_different_order(self) -> None:
-        groups1 = [
-            {"category": "lib", "values": ["matplotlib"]},
-            {"category": "plot", "values": ["scatter"]},
-        ]
-        groups2 = [
-            {"category": "plot", "values": ["scatter"]},
-            {"category": "lib", "values": ["matplotlib"]},
-        ]
+        groups1 = [{"category": "lib", "values": ["matplotlib"]}, {"category": "plot", "values": ["scatter"]}]
+        groups2 = [{"category": "plot", "values": ["scatter"]}, {"category": "lib", "values": ["matplotlib"]}]
         assert _build_cache_key(groups1) == _build_cache_key(groups2)
 
 
@@ -374,10 +359,7 @@ def test_no_filters_returns_all(self) -> None:
         assert len(result) == 1
 
     def test_filter_by_lib(self) -> None:
-        images = [
-            {"spec_id": "s1", "library": "matplotlib"},
-            {"spec_id": "s1", "library": "seaborn"},
-        ]
+        images = [{"spec_id": "s1", "library": "matplotlib"}, {"spec_id": "s1", "library": "seaborn"}]
         spec_lookup = {"s1": {"tags": {}}}
         impl_lookup = {}
         groups = [{"category": "lib", "values": ["matplotlib"]}]
@@ -432,10 +414,7 @@ class TestCalculateOrCounts:
 
     def test_or_counts_for_single_group(self) -> None:
         filter_groups = [{"category": "lib", "values": ["matplotlib"]}]
-        all_images = [
-            {"spec_id": "s1", "library": "matplotlib"},
-            {"spec_id": "s1", "library": "seaborn"},
-        ]
+        all_images = [{"spec_id": "s1", "library": "matplotlib"}, {"spec_id": "s1", "library": "seaborn"}]
         spec_id_to_tags = {"s1": {}}
         spec_lookup = {"s1": {"tags": {}}}
         impl_lookup = {}

tests/unit/api/test_schemas.py

@@ -139,20 +139,15 @@ def test_defaults_empty(self) -> None:
         assert counts.style == {}
 
     def test_with_counts(self) -> None:
-        counts = FilterCountsResponse(
-            lib={"matplotlib": 5, "seaborn": 3},
-            plot={"scatter": 8},
-        )
+        counts = FilterCountsResponse(lib={"matplotlib": 5, "seaborn": 3}, plot={"scatter": 8})
         assert counts.lib["matplotlib"] == 5
 
 
 class TestFilteredPlotsResponse:
     """Tests for FilteredPlotsResponse schema."""
 
     def test_minimal(self) -> None:
-        resp = FilteredPlotsResponse(
-            total=0, images=[], counts={}, globalCounts={}, orCounts=[]
-        )
+        resp = FilteredPlotsResponse(total=0, images=[], counts={}, globalCounts={}, orCounts=[])
         assert resp.total == 0
         assert resp.offset == 0
         assert resp.limit is None

tests/unit/api/test_seo_helpers.py

@@ -140,8 +140,6 @@ def test_template_has_required_meta_tags(self) -> None:
 
     def test_template_has_canonical(self) -> None:
         url = "https://pyplots.ai/"
-        result = BOT_HTML_TEMPLATE.format(
-            title="t", description="d", image="i", url=url
-        )
+        result = BOT_HTML_TEMPLATE.format(title="t", description="d", image="i", url=url)
         assert 'rel="canonical"' in result
         assert url in result

tests/unit/core/database/test_models.py

@@ -7,14 +7,7 @@
 import pytest
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from core.database.models import (
-    Impl,
-    Library,
-    MAX_LIBRARY_ID_LENGTH,
-    MAX_SPEC_ID_LENGTH,
-    REVIEW_VERDICTS,
-    Spec,
-)
+from core.database.models import MAX_LIBRARY_ID_LENGTH, MAX_SPEC_ID_LENGTH, REVIEW_VERDICTS, Impl, Library, Spec
 
 
 class TestModelConstants:
@@ -66,6 +59,7 @@ async def test_persist_and_retrieve(self, test_session: AsyncSession) -> None:
         await test_session.commit()
 
         from sqlalchemy import select
+
         result = await test_session.execute(select(Spec).where(Spec.id == "test-spec"))
         retrieved = result.scalar_one()
         assert retrieved.title == "Test Spec"
@@ -140,16 +134,12 @@ async def test_persist_with_foreign_keys(self, test_session: AsyncSession) -> No
         test_session.add_all([lib, spec])
         await test_session.commit()
 
-        impl = Impl(
-            spec_id="scatter-basic",
-            library_id="matplotlib",
-            code="import matplotlib",
-            quality_score=90.0,
-        )
+        impl = Impl(spec_id="scatter-basic", library_id="matplotlib", code="import matplotlib", quality_score=90.0)
         test_session.add(impl)
         await test_session.commit()
 
         from sqlalchemy import select
+
         result = await test_session.execute(select(Impl).where(Impl.spec_id == "scatter-basic"))
         retrieved = result.scalar_one()
         assert retrieved.quality_score == 90.0