From e5f43ed8100ac99cf685339643f521d0e7e0bced Mon Sep 17 00:00:00 2001 From: joseph-sentry Date: Wed, 13 Nov 2024 16:04:24 -0500 Subject: [PATCH] fix: replace NaN values in aggregates with 0 --- graphql_api/types/flake_aggregates/flake_aggregates.py | 2 +- .../types/test_results_aggregates/test_results_aggregates.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/graphql_api/types/flake_aggregates/flake_aggregates.py b/graphql_api/types/flake_aggregates/flake_aggregates.py index 196c352c04..3d54ccae78 100644 --- a/graphql_api/types/flake_aggregates/flake_aggregates.py +++ b/graphql_api/types/flake_aggregates/flake_aggregates.py @@ -42,7 +42,7 @@ def flake_aggregates_with_percentage( merged_results: pl.DataFrame = pl.concat([past_aggregates, curr_aggregates]) merged_results = merged_results.with_columns( - pl.all().pct_change().name.suffix("_percent_change") + pl.all().pct_change().fill_nan(0).name.suffix("_percent_change") ) aggregates = merged_results.row(1, named=True) diff --git a/graphql_api/types/test_results_aggregates/test_results_aggregates.py b/graphql_api/types/test_results_aggregates/test_results_aggregates.py index 72da5ff7c5..47b73d9866 100644 --- a/graphql_api/types/test_results_aggregates/test_results_aggregates.py +++ b/graphql_api/types/test_results_aggregates/test_results_aggregates.py @@ -71,7 +71,7 @@ def test_results_aggregates_with_percentage( # with_columns upserts the new columns, so if the name already exists it get overwritten # otherwise it's just added merged_results = merged_results.with_columns( - pl.all().pct_change().name.suffix("_percent_change") + pl.all().pct_change().fill_nan(0).name.suffix("_percent_change") ) aggregates = merged_results.row(1, named=True)