Skip to content

Commit e21ab5b

Browse files
committed
Fix ML data_loader: cast Decimal to numeric before interpolate
1 parent 020af49 commit e21ab5b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/ml/data_loader.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ def load_metric_series(db: Session, user_id: int, metric: str) -> pd.DataFrame:
5252

5353
df = pd.DataFrame(rows, columns=["date", "value"])
5454
df["date"] = pd.to_datetime(df["date"])
55+
df["value"] = pd.to_numeric(df["value"], errors="coerce")
5556
df = df.set_index("date").asfreq("D")
5657
df["value"] = df["value"].interpolate(method="linear", limit=3)
5758
return df.dropna()

0 commit comments

Comments
 (0)