nathan/go-weatherstation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix pandas warning messages

Browse Source
This commit is contained in:
Nathan Coad
2026-03-05 11:47:59 +11:00
parent c8e38cd597
commit 76270e5650
1 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
scripts/rain_model_common.py
View File

@@ -49,7 +49,16 @@ def fetch_ws90(conn, site: str, start: str, end: str) -> pd.DataFrame:
AND (%s = '' OR ts <= %s::timestamptz) AND (%s = '' OR ts <= %s::timestamptz)
ORDER BY ts ASC ORDER BY ts ASC
""" """
return pd.read_sql_query(sql, conn, params=(site, start, start, end, end), parse_dates=["ts", "received_at"]) with conn.cursor() as cur:
cur.execute(sql, (site, start, start, end, end))
rows = cur.fetchall()
cols = [d.name for d in cur.description]
df = pd.DataFrame.from_records(rows, columns=cols)
if not df.empty:
df["ts"] = pd.to_datetime(df["ts"], utc=True)
df["received_at"] = pd.to_datetime(df["received_at"], utc=True)
return df
def fetch_baro(conn, site: str, start: str, end: str) -> pd.DataFrame: def fetch_baro(conn, site: str, start: str, end: str) -> pd.DataFrame:
@@ -61,7 +70,16 @@ def fetch_baro(conn, site: str, start: str, end: str) -> pd.DataFrame:
AND (%s = '' OR ts <= %s::timestamptz) AND (%s = '' OR ts <= %s::timestamptz)
ORDER BY ts ASC ORDER BY ts ASC
""" """
return pd.read_sql_query(sql, conn, params=(site, start, start, end, end), parse_dates=["ts", "received_at"]) with conn.cursor() as cur:
cur.execute(sql, (site, start, start, end, end))
rows = cur.fetchall()
cols = [d.name for d in cur.description]
df = pd.DataFrame.from_records(rows, columns=cols)
if not df.empty:
df["ts"] = pd.to_datetime(df["ts"], utc=True)
df["received_at"] = pd.to_datetime(df["received_at"], utc=True)
return df
def build_dataset( def build_dataset(