style fixes

UCL · Jun 10, 2024 · 39f557b · 39f557b
1 parent 5661860
commit 39f557b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/hivpy/hiv_status.py b/src/hivpy/hiv_status.py
@@ -262,7 +262,7 @@ def stp_HIV_transmission(self, person):
 
     def ltp_transmission(self, population: Population):
         """
-        Sets population set for which HIV transmission occurs from a long term partner. 
+        Sets population set for which HIV transmission occurs from a long term partner.
         Considers monogamous and non-monogamous relationships.
         """
         num_neg_women_with_infected_parter = len(population.get_sub_pop([(col.SEX, op.eq, SexType.Female),
@@ -443,7 +443,6 @@ def update_HIV_status(self, population: Population):
         # TODO: Add ltp HIV transmission
         # Determine HIV status after transmission
         new_HIV_status = population.apply_function(self.stp_HIV_transmission, 1, HIV_neg_active_pop)
-
         # Apply HIV status to sub-population
         population.set_present_variable(col.HIV_STATUS,
                                         new_HIV_status,

diff --git a/src/tests/test_hiv_status.py b/src/tests/test_hiv_status.py
@@ -408,7 +408,7 @@ def test_no_infection_in_nonmonogamous_ltp():
 
 def test_ltp_deterministic_variables():
     """
-    Check that for a specific group in the population the estimation is as expected 
+    Check that for a specific group in the population the estimation is as expected
     """
     N = 1000
     pop = Population(size=N, start_date=date(1990, 1, 1))