From 4d4fb2f0dcb73082adb8fc9b65bfba4ac719be63 Mon Sep 17 00:00:00 2001 From: Samyak S Sarnayak Date: Fri, 23 Sep 2022 13:50:07 +0530 Subject: [PATCH] fix(core): replace missing yhats with None/null In some cases `df[col`] can be `np.nan`, so explicitly replace those values with None. The type checker does not like this for some reason. --- chaos_genius/core/utils/round.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chaos_genius/core/utils/round.py b/chaos_genius/core/utils/round.py index 70905822..c8ac28e2 100644 --- a/chaos_genius/core/utils/round.py +++ b/chaos_genius/core/utils/round.py @@ -63,5 +63,5 @@ def round_column_in_df(df: pd.DataFrame, col: str): ), ), ), - df[col] + None, )