black

Author: yamilbknsu

demos/config.py

@@ -186,9 +186,7 @@ class DEMOSConfig(BaseModel):
     kids_moving_module_config: KidsMovingModuleConfig = Field(
         default_factory=KidsMovingModuleConfig
     )
-    income_module_config: IncomeModuleConfig = Field(
-        default_factory=IncomeModuleConfig
-    )
+    income_module_config: IncomeModuleConfig = Field(default_factory=IncomeModuleConfig)
 
     def model_post_init(self, __context) -> None:
         if self.output_fname is None:

demos/models/household_reorg.py

@@ -851,7 +851,9 @@ def update_cohabitating_households(
         households.local.loc[new_households, module_config.geoid_col] = geoid_assignment
     county_assignment = households.local.loc[old_household_id, "lcm_county_id"].values
     households.local.loc[new_households, "lcm_county_id"] = county_assignment
-    not_met_area_assignment = households.local.loc[old_household_id, "not_met_area"].values
+    not_met_area_assignment = households.local.loc[
+        old_household_id, "not_met_area"
+    ].values
     households.local.loc[new_households, "not_met_area"] = not_met_area_assignment
 
 

demos/models/income.py

@@ -32,9 +32,11 @@ def run_and_calibrate_income_model(households):
 
     # Calibrate if needed
     if income_config.calibration_procedure is not None:
-        predicted = np.exp(income_config.calibration_procedure.calibrate_and_run_model(
-            model, model_data
-        ))
+        predicted = np.exp(
+            income_config.calibration_procedure.calibrate_and_run_model(
+                model, model_data
+            )
+        )
     else:
         predicted = np.exp(model.predict(model_data))
 
@@ -107,9 +109,7 @@ def _apply_inflation_adjustment(
             f"years: {missing}.  Ensure the table covers all simulation years."
         )
 
-    cumulative_factor = float(
-        (1 + adj_table.loc[list(years), "adjustment"]).prod()
-    )
+    cumulative_factor = float((1 + adj_table.loc[list(years), "adjustment"]).prod())
     if not forward:
         cumulative_factor = 1.0 / cumulative_factor
 

demos/models/kids_moving.py

@@ -126,7 +126,9 @@ def update_households_after_kids(persons, households, kids_moving, get_new_house
     ].values
     county_assignment = households.local.loc[old_household_id, "lcm_county_id"].values
 
-    not_met_area_assignment = households.local.loc[old_household_id, "not_met_area"].values
+    not_met_area_assignment = households.local.loc[
+        old_household_id, "not_met_area"
+    ].values
 
     new_households = get_new_households(kids_moving_index.sum())
     persons.local.loc[kids_moving_index, "household_id"] = new_households

demos/models/marriage.py

@@ -310,7 +310,9 @@ def update_divorce(persons, households, divorce_list, get_new_households):
         households.local.loc[new_households, module_config.geoid_col] = geoid_assignment
     county_assignment = households.local.loc[old_household_id, "lcm_county_id"].values
     households.local.loc[new_households, "lcm_county_id"] = county_assignment
-    not_met_area_assignment = households.local.loc[old_household_id, "not_met_area"].values
+    not_met_area_assignment = households.local.loc[
+        old_household_id, "not_met_area"
+    ].values
     households.local.loc[new_households, "not_met_area"] = not_met_area_assignment
 
     ## Updates for people staying