minor cleanup

src/vivarium_inputs/utilities.py

@@ -356,7 +356,6 @@ def filter_data_by_restrictions(
 
     start, end = get_age_group_ids_by_restriction(entity, which_age)
     ages = get_restriction_age_ids(start, end, age_group_ids)
-    # ages = ages + [238, 388, 389]
     data = data[data.age_group_id.isin(ages)]
     return data
 
@@ -400,7 +399,9 @@ def get_restriction_age_ids(
     """Get the start/end age group id and return the list of GBD age_group_ids
     in-between.
     """
+    # TODO: remove after MIC-4519 is done
     start_id = 238 if start_id == 4 else start_id
+
     if start_id is None or end_id is None:
         data = []
     else:

src/vivarium_inputs/validation/raw.py

@@ -2051,7 +2051,6 @@ def _check_continuity(data_ages: set, all_ages: set) -> None:
     all_ages = list(all_ages)
     all_ages.sort()
     data_ages.sort()
-
     if (
         all_ages[all_ages.index(data_ages[0]) : all_ages.index(data_ages[-1]) + 1]
         != data_ages

src/vivarium_inputs/validation/sim.py

@@ -1570,7 +1570,9 @@ def check_age_restrictions(
 
     """
     start_id, end_id = utilities.get_age_group_ids_by_restriction(entity, rest_type)
+    # TODO: remove after MIC-4519 is done
     start_id = 238 if start_id==4 else start_id
+
     age_bins = context["age_bins"]
     id_to_age_start_map = dict(zip(age_bins.age_group_id, age_bins.age_start))
     age_range_start = id_to_age_start_map[start_id]