[ci] Re-enable fail on warning for pytest pipeline. (#265)

* [ci] Re-enable fail on warning for pytest pipeline.

* [Fix] Use sqlite3 context managers in pandas module.

* [Fix] Add closing context.
This commit is contained in:
Fabian Joswig 2025-05-05 17:09:40 +02:00 committed by GitHub
parent 8183ee2ef4
commit d6e6a435a8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 7 deletions

View file

@ -43,4 +43,4 @@ jobs:
uv pip freeze --system
- name: Run tests
run: pytest --cov=pyerrors -vv
run: pytest --cov=pyerrors -vv -Werror

View file

@ -1,6 +1,7 @@
import warnings
import gzip
import sqlite3
from contextlib import closing
import pandas as pd
from ..obs import Obs
from ..correlators import Corr
@ -29,9 +30,8 @@ def to_sql(df, table_name, db, if_exists='fail', gz=True, **kwargs):
None
"""
se_df = _serialize_df(df, gz=gz)
con = sqlite3.connect(db)
se_df.to_sql(table_name, con, if_exists=if_exists, index=False, **kwargs)
con.close()
with closing(sqlite3.connect(db)) as con:
se_df.to_sql(table_name, con=con, if_exists=if_exists, index=False, **kwargs)
def read_sql(sql, db, auto_gamma=False, **kwargs):
@ -52,9 +52,8 @@ def read_sql(sql, db, auto_gamma=False, **kwargs):
data : pandas.DataFrame
Dataframe with the content of the sqlite database.
"""
con = sqlite3.connect(db)
extract_df = pd.read_sql(sql, con, **kwargs)
con.close()
with closing(sqlite3.connect(db)) as con:
extract_df = pd.read_sql(sql, con=con, **kwargs)
return _deserialize_df(extract_df, auto_gamma=auto_gamma)