fix picklist pt 3

team4099 · Mar 3, 2024 · 7a9eea3 · 7a9eea3
1 parent f39b437
commit 7a9eea3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/page_managers/picklist_manager.py b/src/page_managers/picklist_manager.py
@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 from notion_client import Client
 from notion_client.helpers import get_id
-from pandas import DataFrame
+from pandas import DataFrame, notna
 
 from .page_manager import PageManager
 from utils import CalculatedStats, Criteria, EventSpecificConstants, Queries, retrieve_scouting_data, retrieve_team_list
@@ -281,7 +281,7 @@ def write_to_notion(self, dataframe: DataFrame) -> None:
                     parent={"type": "database_id", "database_id": db_id},
                     properties={
                         column: {
-                            "number": data if (data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
+                            "number": data if notna(data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
                         } for column in dataframe.columns if column != "Team Number"
                     } | {
                         "Team Name": {"id": "title", "title": [{"text": {"content": team_name}}]},
@@ -316,7 +316,7 @@ def write_to_notion(self, dataframe: DataFrame) -> None:
                     parent={"type": "database_id", "database_id": db_id},
                     properties={
                        column: {
-                           "number": data if (data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
+                           "number": data if notna(data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
                        } for column in dataframe.columns if column != "Team Number"
                     } | {
                        "Team Name": {"id": "title", "title": [{"text": {"content": team_name}}]},