Skip to content

Commit f7d44ba

Browse files
committed
run black
1 parent e29ae8f commit f7d44ba

File tree

2 files changed

+4
-12
lines changed

2 files changed

+4
-12
lines changed

demos/models/household_reorg.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -708,15 +708,9 @@ def update_cohabitating_households(
708708
geoid_assignment = households.local.loc[
709709
old_household_id, module_config.geoid_col
710710
].values
711-
households.local.loc[new_households, module_config.geoid_col] = (
712-
geoid_assignment
713-
)
714-
county_assignment = households.local.loc[
715-
old_household_id, "lcm_county_id"
716-
].values
717-
households.local.loc[new_households, "lcm_county_id"] = (
718-
county_assignment
719-
)
711+
households.local.loc[new_households, module_config.geoid_col] = geoid_assignment
712+
county_assignment = households.local.loc[old_household_id, "lcm_county_id"].values
713+
households.local.loc[new_households, "lcm_county_id"] = county_assignment
720714

721715

722716
def fix_erroneous_households(persons):

demos/models/kids_moving.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -130,9 +130,7 @@ def update_households_after_kids(persons, households, kids_moving, get_new_house
130130
geoid_assignment = households.local.loc[
131131
old_household_id, module_config.geoid_col
132132
].values
133-
county_assignment = households.local.loc[
134-
old_household_id, "lcm_county_id"
135-
].values
133+
county_assignment = households.local.loc[old_household_id, "lcm_county_id"].values
136134

137135
new_households = get_new_households(kids_moving_index.sum())
138136
persons.local.loc[kids_moving_index, "household_id"] = new_households

0 commit comments

Comments
 (0)