linter

Author: Arsalanjdev

src/rowgen/main.py

@@ -1,11 +1,11 @@
-from sqlalchemy import create_engine, text
-
-from rowgen.hf_api import HFapi
 import argparse
+import os
+
+from sqlalchemy import create_engine, text
 
 from rowgen.extract_from_db import DBconnect
+from rowgen.hf_api import HFapi
 from rowgen.sql_parser import parse_sql_from_code_block
-import os
 
 
 def get_parser():

src/rowgen/utils.py

@@ -1,7 +1,7 @@
+import os
 from pathlib import Path
 
 from dotenv import load_dotenv
-import os
 
 
 def find_project_root(start_path: Path = None, marker_files=None) -> Path:

tests/test_ai_api.py

@@ -1,13 +1,9 @@
-import os
-
 import pytest
-from dotenv import load_dotenv
+from huggingface_hub import HfApi, InferenceClient
 from huggingface_hub.errors import BadRequestError
 
-from rowgen.utils import API_KEY
-from huggingface_hub import InferenceClient, HfApi
 from rowgen.hf_api import HFapi
-
+from rowgen.utils import API_KEY
 
 DEEP_SEEK_MODEL = "deepseek-ai/DeepSeek-V3"
 

tests/test_argument_parser.py

@@ -1,5 +1,7 @@
 import sys
+
 import pytest
+
 from rowgen.main import get_parser
 
 

tests/test_db.py

@@ -1,9 +1,9 @@
 import os
+import sqlite3
 
 import pytest
-from sqlalchemy import create_engine, INTEGER, TEXT
-from sqlalchemy.orm import sessionmaker
-import sqlite3
+from sqlalchemy import INTEGER, TEXT
+
 from rowgen.extract_from_db import DBconnect
 
 
@@ -37,7 +37,7 @@ def db_connect(temp_db):
 
 
 def test_get_columns(db_connect):
-    cols = db_connect.table_columns
+    # cols = db_connect.table_columns
     expected = {
         "users": [
             {