repair input check

jpn-- · jpn-- · commit 5f27067a07c0 · 2024-02-13T23:28:58.000-06:00
diff --git a/data_model/input_checks.py b/data_model/input_checks.py
@@ -230,7 +230,8 @@ def check_pop_per_zone(cls, land_use: pd.DataFrame):
         pop = (
             hh.groupby(households.home_zone_id)["persons_per_household"].sum()
         )
-        return (pop == land_use.set_index("zone_id").TOTPOP).reindex(land_use.index)
+        lu = land_use.set_index("zone_id")
+        return (pop.reindex(lu.index) == lu.TOTPOP)
 
 
 class NetworkLinks(pa.DataFrameModel):

Original file line number	Diff line number	Diff line change
`@@ -230,7 +230,8 @@ def check_pop_per_zone(cls, land_use: pd.DataFrame):`
`230`	`230`	`pop = (`
`231`	`231`	`hh.groupby(households.home_zone_id)["persons_per_household"].sum()`
`232`	`232`	`)`
`233`		`- return (pop == land_use.set_index("zone_id").TOTPOP).reindex(land_use.index)`
	`233`	`+ lu = land_use.set_index("zone_id")`
	`234`	`+ return (pop.reindex(lu.index) == lu.TOTPOP)`
`234`	`235`
`235`	`236`
`236`	`237`	`class NetworkLinks(pa.DataFrameModel):`