Lint

ihmeuw · Nov 14, 2024 · ce5bc78 · ce5bc78
1 parent f277568
commit ce5bc78
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/vivarium_gates_mncnh/components/antenatal_care.py b/src/vivarium_gates_mncnh/components/antenatal_care.py
@@ -24,7 +24,6 @@
 
 
 class TreeMachine(Machine):
-
     def setup(self, builder: Builder) -> None:
         super().setup(builder)
         self._sim_step_name = builder.time.simulation_event_name()
@@ -191,8 +190,8 @@ def create_anc_decision_tree(self) -> Machine:
             probability_function=self.get_anc_attendance_rate,
         )
         initial_state.add_transition(
-            output_state=end_state, 
-            probability_function=lambda index: 1 - self.get_anc_attendance_rate(index)
+            output_state=end_state,
+            probability_function=lambda index: 1 - self.get_anc_attendance_rate(index),
         )
         attended_antental_care.add_transition(
             output_state=gets_ultrasound,