fix: resolve all ruff lint errors

- Fix E402 in cli/app.py: move lazy imports into function bodies
- Fix E501 in data_analysis.py: break long lines in template code
- Auto-fix 16 errors: unused imports, unsorted imports in test files

Author: dominikpeter

retrai/agent/nodes/plan.py

@@ -12,7 +12,6 @@
 from retrai.agent.state import AgentState, ToolCall
 from retrai.events.types import AgentEvent
 from retrai.llm.factory import get_llm
-
 from retrai.tools.builtins import create_default_registry
 
 logger = logging.getLogger(__name__)

retrai/cli/app.py

@@ -351,16 +351,12 @@ def run(
         )
     )
 
+    from retrai.cli.runners import run_cli as _run_cli
+
     exit_code = asyncio.run(_run_cli(cfg))
     raise typer.Exit(code=exit_code)
 
 
-from retrai.cli.runners import (
-    run_cli as _run_cli,
-    run_solve as _run_solve,
-    run_swarm as _run_swarm,
-)
-
 
 @app.command()
 def serve(
@@ -753,6 +749,8 @@ def solve(
         hitl_enabled=False,
     )
 
+    from retrai.cli.runners import run_solve as _run_solve
+
     exit_code = asyncio.run(_run_solve(cfg, description))
     raise typer.Exit(code=exit_code)
 
@@ -814,6 +812,8 @@ def swarm(
         )
     )
 
+    from retrai.cli.runners import run_swarm as _run_swarm
+
     exit_code = asyncio.run(_run_swarm(
         description=description,
         cwd=resolved_cwd,

retrai/tools/builtins.py

@@ -14,7 +14,6 @@
 
 from retrai.tools.base import BaseTool, ToolRegistry, ToolSchema
 
-
 # ──────────────────────────────────────────────────────────────────
 # Shell / Execution
 # ──────────────────────────────────────────────────────────────────

retrai/tools/data_analysis.py

@@ -5,7 +5,6 @@
 import json
 import logging
 import textwrap
-from typing import Any
 
 from retrai.tools.python_exec import python_exec
 
@@ -93,7 +92,10 @@ def _build_summary_code(file_path: str) -> str:
             "columns": list(df.columns),
             "dtypes": {{col: str(dtype) for col, dtype in df.dtypes.items()}},
             "missing": {{col: int(v) for col, v in df.isnull().sum().items() if v > 0}},
-            "missing_pct": {{col: round(v / len(df) * 100, 1) for col, v in df.isnull().sum().items() if v > 0}},
+            "missing_pct": {{
+                col: round(v / len(df) * 100, 1)
+                for col, v in df.isnull().sum().items() if v > 0
+            }},
         }}
 
         # Numeric summary
@@ -181,7 +183,10 @@ def _build_quality_code(file_path: str) -> str:
             "memory_mb": round(df.memory_usage(deep=True).sum() / 1024 / 1024, 2),
             "duplicate_rows": int(df.duplicated().sum()),
             "complete_rows": int(df.dropna().shape[0]),
-            "completeness_pct": round(df.dropna().shape[0] / len(df) * 100, 1) if len(df) > 0 else 0,
+            "completeness_pct": (
+                round(df.dropna().shape[0] / len(df) * 100, 1)
+                if len(df) > 0 else 0
+            ),
         }}
 
         # Per-column quality
@@ -190,9 +195,15 @@ def _build_quality_code(file_path: str) -> str:
             info = {{
                 "dtype": str(df[col].dtype),
                 "missing": int(df[col].isnull().sum()),
-                "missing_pct": round(df[col].isnull().sum() / len(df) * 100, 1) if len(df) > 0 else 0,
+                "missing_pct": (
+                    round(df[col].isnull().sum() / len(df) * 100, 1)
+                    if len(df) > 0 else 0
+                ),
                 "unique": int(df[col].nunique()),
-                "unique_pct": round(df[col].nunique() / len(df) * 100, 1) if len(df) > 0 else 0,
+                "unique_pct": (
+                    round(df[col].nunique() / len(df) * 100, 1)
+                    if len(df) > 0 else 0
+                ),
             }}
             if df[col].dtype in ["float64", "int64"]:
                 info["zeros"] = int((df[col] == 0).sum())
@@ -240,12 +251,21 @@ def _build_distribution_code(file_path: str) -> str:
             col_info = {{"dtype": str(df[col].dtype)}}
 
             if df[col].dtype in ["float64", "int64"]:
-                col_info["min"] = float(df[col].min()) if not df[col].isnull().all() else None
-                col_info["max"] = float(df[col].max()) if not df[col].isnull().all() else None
-                col_info["mean"] = round(float(df[col].mean()), 4) if not df[col].isnull().all() else None
-                col_info["median"] = float(df[col].median()) if not df[col].isnull().all() else None
-                col_info["std"] = round(float(df[col].std()), 4) if not df[col].isnull().all() else None
-                col_info["skewness"] = round(float(df[col].skew()), 4) if not df[col].isnull().all() else None
+                not_null = not df[col].isnull().all()
+                col_info["min"] = float(df[col].min()) if not_null else None
+                col_info["max"] = float(df[col].max()) if not_null else None
+                col_info["mean"] = (
+                    round(float(df[col].mean()), 4) if not_null else None
+                )
+                col_info["median"] = (
+                    float(df[col].median()) if not_null else None
+                )
+                col_info["std"] = (
+                    round(float(df[col].std()), 4) if not_null else None
+                )
+                col_info["skewness"] = (
+                    round(float(df[col].skew()), 4) if not_null else None
+                )
 
                 # Histogram bins
                 try:

tests/unit/test_js_exec.py

@@ -15,7 +15,6 @@
     js_exec,
 )
 
-
 # ── js_sandbox_dir ────────────────────────────────────────────────────────────
 
 

tests/unit/test_ml_train.py

@@ -4,7 +4,6 @@
 
 import json
 from pathlib import Path
-from unittest.mock import AsyncMock, patch
 
 import pytest
 
@@ -16,7 +15,6 @@
     ml_train,
 )
 
-
 # ── ml_train validation ──────────────────────────────────────────────────────
 
 

tests/unit/test_research_goal.py

@@ -4,7 +4,6 @@
 
 import json
 from pathlib import Path
-from unittest.mock import patch
 
 import pytest
 

tests/unit/test_swarm_integration.py

@@ -3,9 +3,6 @@
 from __future__ import annotations
 
 import json
-from unittest.mock import AsyncMock, MagicMock, patch
-
-import pytest
 
 from retrai.swarm.decomposer import _parse_subtasks
 from retrai.swarm.roles import get_role

tests/unit/test_visualize.py

@@ -3,14 +3,12 @@
 from __future__ import annotations
 
 import json
-import textwrap
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 
 from retrai.tools.visualize import VALID_CHART_TYPES, _build_chart_code, visualize
 
-
 # ---------------------------------------------------------------------------
 # _build_chart_code tests
 # ---------------------------------------------------------------------------