PR comment

ihmeuw · Dec 12, 2023 · ccc6413 · ccc6413
1 parent 610f47b
commit ccc6413
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/vivarium_gates_nutrition_optimization/components/intervention.py b/src/vivarium_gates_nutrition_optimization/components/intervention.py
@@ -78,12 +78,12 @@ def on_initialize_simulants(self, pop_data: SimulantData) -> None:
         pop = self.population_view.subview(["maternal_bmi_anemia_category"]).get(
             pop_data.index
         )
+
         if self.scenario == 'ifa':
             pop_update = pd.DataFrame(
                 {"intervention": 'ifa'},
                 index=pop.index,
             )
-            self.population_view.update(pop_update)
         else:
             pop_update = pd.DataFrame(
                 {"intervention": None},
@@ -105,7 +105,8 @@ def on_initialize_simulants(self, pop_data: SimulantData) -> None:
 
             unsampled_ifa = pop_update["intervention"] == "maybe_ifa"
             pop_update.loc[unsampled_ifa, "intervention"] = baseline_ifa.loc[unsampled_ifa]
-            self.population_view.update(pop_update)
+
+        self.population_view.update(pop_update)
 
     def update_exposure(self, index, exposure):
         if self.clock() - self.start_date >= timedelta(