Prevent pandas contains method to crash on NAN values

KlimaDAO · Sep 6, 2023 · 393b003 · 393b003
1 parent 2f40334
commit 393b003
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/apps/tco2_dashboard/helpers.py b/src/apps/tco2_dashboard/helpers.py
@@ -343,7 +343,7 @@ def merge_retirements_data_for_retirement_chart(
         "Beneficiary",
     ] = df_retired_polygon_merged["Tx From Address"]
     df_retired_polygon_merged.loc[
-        df_retired_polygon_merged["Beneficiary"].str.contains("Polygon", case=False),
+        df_retired_polygon_merged["Beneficiary"].str.contains("Polygon", case=False, na=False),
         "Beneficiary",
     ] = "Polygon"
 
@@ -353,7 +353,7 @@ def merge_retirements_data_for_retirement_chart(
     ] = df_retired_polygon_merged["Quantity_klima_agg"]
 
     df_retired_polygon_merged.loc[
-        df_retired_polygon_merged["Beneficiary"].str.contains("Polygon", case=False),
+        df_retired_polygon_merged["Beneficiary"].str.contains("Polygon", case=False, na=False),
         "Beneficiary",
     ] = "Polygon"
 
@@ -388,11 +388,11 @@ def merge_retirements_data_for_retirement_chart(
 
     # Prettify known beneficiaries
     df_retired_eth_merged.loc[
-        df_retired_eth_merged["Beneficiary"].str.contains("ifood|IFOOD", case=False),
+        df_retired_eth_merged["Beneficiary"].str.contains("ifood|IFOOD", case=False, na=False),
         "Beneficiary",
     ] = "iFood"
     df_retired_eth_merged.loc[
-        df_retired_eth_merged["Beneficiary"].str.contains("SKYBRIDGE", case=False),
+        df_retired_eth_merged["Beneficiary"].str.contains("SKYBRIDGE", case=False, na=False),
         "Beneficiary",
     ] = "SkyBridge Capital"