diff --git a/backend/app.py b/backend/app.py index e50dd50..08895be 100644 --- a/backend/app.py +++ b/backend/app.py @@ -827,10 +827,10 @@ async def get_data_info(request: Request): "data_source": "PostgreSQL", } - years = [int(y) for y in sorted(df["year"].unique())] + years = [int(y) for y in sorted(df["year"].dropna().unique())] schools_per_year = { str(int(k)): int(v) - for k, v in df.groupby("year")["urn"].nunique().to_dict().items() + for k, v in df.dropna(subset=["year"]).groupby("year")["urn"].nunique().to_dict().items() } la_counts = { str(k): int(v)