style: fix tzinfo issues with ruff

Author: dvp

pyproject.toml

@@ -48,9 +48,9 @@ Changelog = "https://github.com/MC-kit/xpypact/releases"
 
 
 [tool.poetry.dependencies]
-# msgspec-0.18.5 doesn't work on 3.9 - uses | without importing annotations
+# msgspec-0.18.5 doesn't work on 3.9 - uses `|` without importing annotations from __future__
 # duckdb-0.9.2, has no wheels for 3.12 and fails to build from source
-python = ">=3.10,<3.13"
+python = ">=3.9,<3.13"
 duckdb = ">=0.8.0"
 # mckit-nuclides = {version = ">=0.2.5", allow-prereleases = true}
 numpy = ">=1.26.0"

src/xpypact/collector.py

@@ -4,7 +4,7 @@
 
 from typing import TYPE_CHECKING
 
-import datetime
+import datetime as dt
 import sys
 import threading
 
@@ -26,9 +26,9 @@
 
 
 if sys.version_info >= (3, 11):  # pragma: no cover
-    UTC = datetime.UTC
+    UTC = dt.UTC
 else:
-    UTC = datetime.timezone.utc  # pragma: no cover
+    UTC = dt.timezone.utc  # pragma: no cover
 
 RunDataSchema = OrderedDict(
     material_id=pl.UInt32,
@@ -147,14 +147,14 @@ def append(self, inventory: Inventory, material_id: int, case_id: int) -> FullDa
     def _append_rundata(self, inventory, material_id, case_id):
         rundata = inventory.meta_info
         st = strptime(rundata.timestamp, "%H:%M:%S %d %B %Y")
-        ts = datetime.datetime(
+        ts = dt.datetime(  # noqa: DTZ001 - no tzinfo is available from the FISPACT output
             year=st.tm_year,
             month=st.tm_mon,
             day=st.tm_mday,
             hour=st.tm_hour,
             minute=st.tm_min,
             second=st.tm_sec,
-            tzinfo=UTC,
+            tzinfo=None,
         )
         rundata_df = pl.DataFrame(
             [

src/xpypact/dao/duckdb/implementation.py

@@ -10,7 +10,6 @@
 
 if TYPE_CHECKING:
     import duckdb as db
-    import pandas as pd
 
     from xpypact.collector import FullDataCollector
 
@@ -30,9 +29,9 @@ class DuckDBDAO(ms.Struct):
 
     con: db.DuckDBPyConnection
 
-    def get_tables_info(self) -> pd.DataFrame:
+    def get_tables_info(self) -> db.DuckDBPyRelation:
         """Get information on tables in schema."""
-        return self.con.execute("select * from information_schema.tables").df()
+        return self.con.sql("select * from information_schema.tables")
 
     def tables(self) -> tuple[str, str, str, str, str]:
         """List tables being used by xpypact dao.
@@ -44,13 +43,11 @@ def tables(self) -> tuple[str, str, str, str, str]:
 
     def has_schema(self) -> bool:
         """Check if the schema is available in a database."""
-        db_tables = self.get_tables_info()
+        table_names = self.get_tables_info().select("table_name").fetchnumpy()["table_name"]
 
-        if len(db_tables) < len(self.tables()):
+        if len(table_names) < len(self.tables()):
             return False
 
-        table_names = db_tables["table_name"].to_numpy()
-
         return all(name in table_names for name in self.tables())
 
     def create_schema(self) -> None:

tests/test_duckdb_dao.py

@@ -10,7 +10,7 @@
 import pytest
 
 from duckdb import InvalidInputException, connect
-from xpypact.collector import UTC, FullDataCollector
+from xpypact.collector import FullDataCollector
 from xpypact.dao.duckdb import DuckDBDAO as DataAccessObject
 from xpypact.dao.duckdb import create_indices
 from xpypact.dao.duckdb.implementation import save
@@ -48,14 +48,15 @@ def test_save(inventory_with_gamma) -> None:
         dc.append(inventory_with_gamma, material_id=1, case_id=1)
         save(con, dc.get_result())
         run_data = dao.load_rundata().pl()
-        assert run_data.select("timestamp").item() == dt.datetime(
+        assert run_data.select(
+            "timestamp",
+        ).item() == dt.datetime(  # noqa: DTZ001 - no tzinfo is available
             2022,
             2,
             21,
             1,
             52,
             45,
-            tzinfo=UTC,
         )
         assert run_data["run_name"].item() == "* Material Cu, fluxes 104_2_1_1"
         nuclides = dao.load_nuclides().pl()