Merge pull request #493 from chaos-genius/rca-nan-mean-fix

fixes #494 where NaN output was found in deepdrills for specific case
chaos-genius · Dec 9, 2021 · 2f6758d · 2f6758d
2 parents c04badb + 78ad6de
commit 2f6758d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/chaos_genius/core/rca/root_cause_analysis.py b/chaos_genius/core/rca/root_cause_analysis.py
@@ -360,6 +360,8 @@ def _get_waterfall_output_data(
 
         d1_agg = self._grp1_df[self._metric].agg(self._agg)
         d2_agg = self._grp2_df[self._metric].agg(self._agg)
+        d1_agg = 0 if pd.isna(d1_agg) else d1_agg
+        d2_agg = 0 if pd.isna(d2_agg) else d2_agg
         impact = d2_agg - d1_agg
         non_overlap_impact = df_subgroups["impact_non_overlap"].sum()