Fix bug caused by trying to iterate over a DictReader more than once.

RTIInternational · Apr 17, 2024 · 6c264ca · 6c264ca
1 parent 55312a0
commit 6c264ca
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/crcsim/__main__.py b/crcsim/__main__.py
@@ -98,7 +98,7 @@ def run(
     out.open()
 
     with open(cohort_file, mode="r") as input:
-        cohort = csv.DictReader(input)
+        cohort = list(csv.DictReader(input))
 
         # Draw expected lifespans for everyone in the cohort prior to starting
         # simulations. This is necessary to ensure that expected lifespans are

diff --git a/crcsim/agent.py b/crcsim/agent.py
@@ -997,7 +997,6 @@ def is_false_positive(self, test: str):
     # on_end_year is just a wrapper for update_value - not necessary as far as I can tell
 
     def start_life_timer(self):
-        self.expected_lifespan = self.compute_lifespan()
         self.scheduler.add_event(
             message=PersonDiseaseMessage.OTHER_DEATH,
             handler=self.handle_disease_message,