lint

ihmeuw · Sep 13, 2024 · f30abea · f30abea
1 parent 968f141
commit f30abea
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 19 deletions.
diff --git a/src/vivarium_sodium_reduction/constants/data_keys.py b/src/vivarium_sodium_reduction/constants/data_keys.py
@@ -33,9 +33,13 @@ class __IschemicHeartDisease(NamedTuple):
     # Keys that will be loaded into the artifact. must have a colon type declaration
     PREVALENCE: TargetString = TargetString("cause.ischemic_heart_disease.prevalence")
     INCIDENCE_RATE: TargetString = TargetString("cause.ischemic_heart_disease.incidence_rate")
-    DISABILITY_WEIGHT: TargetString = TargetString("cause.ischemic_heart_disease.disability_weight")
+    DISABILITY_WEIGHT: TargetString = TargetString(
+        "cause.ischemic_heart_disease.disability_weight"
+    )
     EMR: TargetString = TargetString("cause.ischemic_heart_disease.excess_mortality_rate")
-    CSMR: TargetString = TargetString("cause.ischemic_heart_disease.cause_specific_mortality_rate")
+    CSMR: TargetString = TargetString(
+        "cause.ischemic_heart_disease.cause_specific_mortality_rate"
+    )
     RESTRICTIONS: TargetString = TargetString("cause.ischemic_heart_disease.restrictions")
 
     # Useful keys not for the artifact - distinguished by not using the colon type declaration
@@ -101,10 +105,16 @@ def log_name(self):
 class __IntracerebralHemorrhage(NamedTuple):
     # Keys that will be loaded into the artifact. must have a colon type declaration
     PREVALENCE: TargetString = TargetString("cause.intracerebral_hemorrhage.prevalence")
-    INCIDENCE_RATE: TargetString = TargetString("cause.intracerebral_hemorrhage.incidence_rate")
-    DISABILITY_WEIGHT: TargetString = TargetString("cause.intracerebral_hemorrhage.disability_weight")
+    INCIDENCE_RATE: TargetString = TargetString(
+        "cause.intracerebral_hemorrhage.incidence_rate"
+    )
+    DISABILITY_WEIGHT: TargetString = TargetString(
+        "cause.intracerebral_hemorrhage.disability_weight"
+    )
     EMR: TargetString = TargetString("cause.intracerebral_hemorrhage.excess_mortality_rate")
-    CSMR: TargetString = TargetString("cause.intracerebral_hemorrhage.cause_specific_mortality_rate")
+    CSMR: TargetString = TargetString(
+        "cause.intracerebral_hemorrhage.cause_specific_mortality_rate"
+    )
     RESTRICTIONS: TargetString = TargetString("cause.intracerebral_hemorrhage.restrictions")
 
     # Useful keys not for the artifact - distinguished by not using the colon type declaration
@@ -124,10 +134,16 @@ def log_name(self):
 class __SubarachnoidHemorrhage(NamedTuple):
     # Keys that will be loaded into the artifact. must have a colon type declaration
     PREVALENCE: TargetString = TargetString("cause.subarachnoid_hemorrhage.prevalence")
-    INCIDENCE_RATE: TargetString = TargetString("cause.subarachnoid_hemorrhage.incidence_rate")
-    DISABILITY_WEIGHT: TargetString = TargetString("cause.subarachnoid_hemorrhage.disability_weight")
+    INCIDENCE_RATE: TargetString = TargetString(
+        "cause.subarachnoid_hemorrhage.incidence_rate"
+    )
+    DISABILITY_WEIGHT: TargetString = TargetString(
+        "cause.subarachnoid_hemorrhage.disability_weight"
+    )
     EMR: TargetString = TargetString("cause.subarachnoid_hemorrhage.excess_mortality_rate")
-    CSMR: TargetString = TargetString("cause.subarachnoid_hemorrhage.cause_specific_mortality_rate")
+    CSMR: TargetString = TargetString(
+        "cause.subarachnoid_hemorrhage.cause_specific_mortality_rate"
+    )
     RESTRICTIONS: TargetString = TargetString("cause.subarachnoid_hemorrhage.restrictions")
 
     # Useful keys not for the artifact - distinguished by not using the colon type declaration
@@ -178,12 +194,8 @@ def log_name(self):
 
 
 class __HighSodium(NamedTuple):
-    DISTRIBUTION: TargetString = TargetString(
-        "risk_factor.diet_high_in_sodium.distribution"
-    )
-    EXPOSURE_MEAN: TargetString = TargetString(
-        "risk_factor.diet_high_in_sodium.exposure"
-    )
+    DISTRIBUTION: TargetString = TargetString("risk_factor.diet_high_in_sodium.distribution")
+    EXPOSURE_MEAN: TargetString = TargetString("risk_factor.diet_high_in_sodium.exposure")
     EXPOSURE_SD: TargetString = TargetString(
         "risk_factor.diet_high_in_sodium.exposure_standard_deviation"
     )

diff --git a/src/vivarium_sodium_reduction/constants/models.py b/src/vivarium_sodium_reduction/constants/models.py
@@ -1,12 +1,11 @@
 from vivarium_sodium_reduction.constants import data_keys
 
-
 ###########################
 # Disease Model variables #
 ###########################
 
 # TODO input details of model states
-#SOME_MODEL_NAME = data_keys.SOME_DISEASE.name
-#SUSCEPTIBLE_STATE_NAME = f"susceptible_to_{SOME_MODEL_NAME}"
-#FIRST_STATE_NAME = "first_state"
-#SECOND_STATE_NAME = "second_state"
+# SOME_MODEL_NAME = data_keys.SOME_DISEASE.name
+# SUSCEPTIBLE_STATE_NAME = f"susceptible_to_{SOME_MODEL_NAME}"
+# FIRST_STATE_NAME = "first_state"
+# SECOND_STATE_NAME = "second_state"