Style

UCL · Oct 3, 2024 · 3e42c5f · 3e42c5f
1 parent c516686
commit 3e42c5f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/hivpy/common.py b/src/hivpy/common.py
@@ -59,18 +59,21 @@ class SexType(IntEnum):
 def opposite_sex(sex: SexType):
     return (1 - sex)
 
+
 def sub_pop_ratio(sp1, sp2):
     if len(sp2) == 0:
         return 0
     else:
         return len(sp1) / len(sp2)
 
+
 def safe_ratio(n1, n2):
     if n2 == 0:
         return 0
     else:
         return n1/n2
 
+
 class date:
     def __init__(self, year, month=1, day=1):
         self.year = year

diff --git a/src/hivpy/hiv_status.py b/src/hivpy/hiv_status.py
@@ -13,7 +13,8 @@
 import hivpy.column_names as col
 
 from . import output
-from .common import AND, COND, SexType, opposite_sex, rng, timedelta, safe_ratio
+from .common import (AND, COND, SexType, opposite_sex, rng, safe_ratio,
+                     timedelta)
 
 
 class HIVStatusModule:
@@ -119,7 +120,7 @@ def init_HIV_variables(self, population: Population):
         population.init_variable(col.VIRAL_LOAD, 0.0)
         population.init_variable(col.VIRAL_SUPPRESSION, False)
         population.init_variable(col.X4_VIRUS, False)
-        
+
         # TODO: move to ART module
         population.init_variable(col.ON_ART, False)