Remove all unused imports

Author: nicoloesch

src/orm_loader/helpers/null_handlers.py

@@ -1,4 +1,3 @@
-import math
 import pandas as pd
 from typing import Any
 

src/orm_loader/tables/loadable_table.py

@@ -102,7 +102,7 @@ def create_staging_table(
         dialect = session.bind.dialect.name 
 
         if dialect == "postgresql":
-            logger.info(f"Disabling indices on staging table for performance")
+            logger.info("Disabling indices on staging table for performance")
             session.execute(sa.text(f'''
                 CREATE UNLOGGED TABLE "{cls.staging_tablename()}"
                 (LIKE "{table.name}" INCLUDING DEFAULTS INCLUDING CONSTRAINTS);

tests/conftest.py

@@ -1,8 +1,6 @@
 import pytest
 import sqlalchemy as sa
 import sqlalchemy.orm as so
-import os
-from orm_loader.tables import CSVLoadableTableInterface
 import time
 
 from tests.models import Base

tests/loaders/test_loader_e2e.py

@@ -379,7 +379,7 @@ class TextTable(Base, CSVLoadableTableInterface):
         '1\t"hello\nworld"\n'
     )
 
-    inserted = TextTable.load_csv( # type: ignore
+    TextTable.load_csv( # type: ignore
         session,
         csv,
         loader=PandasLoader(),
@@ -406,7 +406,7 @@ class TextTable2(Base, CSVLoadableTableInterface):
         '1\t"foo\tbar"\n'
     )
 
-    inserted = TextTable2.load_csv( # type: ignore
+    TextTable2.load_csv( # type: ignore
         session,
         csv,
         loader=PandasLoader(),

tests/loaders/test_pandas_loader.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 import sqlalchemy as sa
-import sqlalchemy.orm as so
 from orm_loader.loaders.data_classes import LoaderContext
 import pandas as pd
 from orm_loader.loaders.loader_interface import PandasLoader

tests/loaders/test_pg_loader.py

@@ -1,17 +1,10 @@
 import sqlalchemy as sa
-import sqlalchemy.orm as so
-from sqlalchemy.orm import Session
-from pathlib import Path
 import pandas as pd
 import pytest
-from orm_loader.loaders.data_classes import _clean_nulls
-from orm_loader.tables.loadable_table import CSVLoadableTableInterface
-from orm_loader.loaders.loader_interface import PandasLoader
 from orm_loader.loaders.loading_helpers import infer_encoding, infer_delim, check_line_ending, quick_load_pg
 
-from tests.models import Base, SimpleTable, RequiredTable, CompositeTable
+from tests.models import SimpleTable
 
-import numpy as np
 
 @pytest.mark.postgres
 def test_copy_and_orm_path_equivalence(pg_session, tmp_path):
@@ -24,7 +17,7 @@ def test_copy_and_orm_path_equivalence(pg_session, tmp_path):
         ]
     ).to_csv(csv, index=False, sep="\t")
 
-    inserted = SimpleTable.load_csv(pg_session, csv)
+    SimpleTable.load_csv(pg_session, csv)
     pg_session.commit()
 
     rows = pg_session.execute(sa.select(SimpleTable).order_by(SimpleTable.id)).scalars().all()
@@ -298,7 +291,7 @@ def test_copy_fails_with_raw_carriage_returns_but_succeeds_after_normalisation(p
     try:
         quick_load_pg(path=csv, session=pg_session, tablename="test_table")
         pg_session.commit()
-    except Exception as e:
+    except Exception:
         failed = True
 
     # Clean up table for second attempt

tests/pg_db.py

@@ -1,9 +1,7 @@
-import os
 import time
 import pytest
 import sqlalchemy as sa
 from sqlalchemy.orm import sessionmaker
-from sqlalchemy_utils import create_database, database_exists, drop_database
 
 from tests.models import Base