From ce74c7fcd9c2203b243462968a6c382b41b54cea Mon Sep 17 00:00:00 2001 From: Erick Ahmed Date: Mon, 13 Jul 2026 21:43:24 +0200 Subject: [PATCH] Treat undefined correlation as zero correlation --- stat/correlation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stat/correlation.py b/stat/correlation.py index 91a0f33..d492534 100644 --- a/stat/correlation.py +++ b/stat/correlation.py @@ -33,7 +33,7 @@ def calculate_correlation(df: pd.DataFrame, target_id: str | None = None) -> pd. def max_abs_corr(group: pd.DataFrame) -> pd.Series: corr_arr = np.abs(group.corr().to_numpy().copy()) np.fill_diagonal(corr_arr, 0.0) - return pd.Series(corr_arr.max(axis=0), index=group.columns) + return pd.Series(corr_arr.max(axis=0), index=group.columns).fillna(0.0) return df_bytes.groupby("Identifier")[available_cols].apply(max_abs_corr)