From 70ad58dd7a393d68eab63164963f10a8145aa517 Mon Sep 17 00:00:00 2001
From: AI Assistant <ai@example.com>
Date: Sun, 15 Mar 2026 20:30:15 +0530
Subject: [PATCH] fix: resolve all identified linting errors (unused imports,
 variables, and improperly placed imports)

---
 pipelineprobe/connectors/dbt.py | 1 -
 tests/test_cli.py               | 7 ++-----
 tests/test_connectors.py        | 3 +--
 tests/test_renderer.py          | 2 +-
 4 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/pipelineprobe/connectors/dbt.py b/pipelineprobe/connectors/dbt.py
index 2ce3dd5..1ef885a 100644
--- a/pipelineprobe/connectors/dbt.py
+++ b/pipelineprobe/connectors/dbt.py
@@ -13,7 +13,6 @@ def __init__(self, config: DbtConfig):
         self.config = config
 
     def get_models(self) -> List[DbtModel]:
-        project_dir = Path(self.config.project_dir)
         manifest_path = Path(self.config.manifest_path)
         run_results_path = Path(self.config.run_results_path)
 
diff --git a/tests/test_cli.py b/tests/test_cli.py
index cbec5c5..4737882 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,8 +1,7 @@
 import os
 import tempfile
-
+from unittest.mock import patch
 from typer.testing import CliRunner
-
 from pipelineprobe.cli import app
 
 runner = CliRunner()
@@ -28,9 +27,7 @@ def test_cli_audit_help():
     assert "Run the PipelineProbe audit and generate a report" in result.stdout
 
 
-import os
-import tempfile
-from unittest.mock import patch
+
 
 def test_cli_audit_full():
     with tempfile.TemporaryDirectory() as tmpdir:
diff --git a/tests/test_connectors.py b/tests/test_connectors.py
index 9ea9d24..c8c7838 100644
--- a/tests/test_connectors.py
+++ b/tests/test_connectors.py
@@ -1,6 +1,5 @@
 import json
-import pytest
-from unittest.mock import AsyncMock, patch, mock_open, MagicMock
+from unittest.mock import patch, MagicMock
 
 from pipelineprobe.config import WarehouseConfig, DbtConfig, AirflowConfig
 from pipelineprobe.connectors.postgres import PostgresConnector
diff --git a/tests/test_renderer.py b/tests/test_renderer.py
index 12cd3ed..00d017e 100644
--- a/tests/test_renderer.py
+++ b/tests/test_renderer.py
@@ -3,7 +3,7 @@
 from datetime import timedelta
 
 from pipelineprobe.renderer import ReportRenderer
-from pipelineprobe.models import Issue, Dag, Task
+from pipelineprobe.models import Issue
 
 def test_render_html(tmp_path):
     renderer = ReportRenderer(output_dir=str(tmp_path))