update-pqts fix

hudcostreets · Feb 3, 2024 · f14aadb · f14aadb
1 parent 984d92c
commit f14aadb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/njsp/cli/update_pqts.py b/njsp/cli/update_pqts.py
@@ -101,7 +101,7 @@ def update_pqts():
 
     # Verify the reported "total deaths" stat reflects what we see in the crash records
     njsp_totals = totals.fatalities.rename('NJSP total')
-    fatalities_per_year = crashes.FATALITIES.groupby(dt.year).sum().astype(int).rename('NJSP records')
+    fatalities_per_year = crashes.FATALITIES.groupby(crashes.dt.dt.year).sum().astype(int).rename('NJSP records')
     njsp_diffs = sxs(njsp_totals, fatalities_per_year)[njsp_totals != fatalities_per_year]
     if not njsp_diffs.empty:
         raise RuntimeError(f"NJSP totals don't match crash records:\n{njsp_diffs}")