Merge pull request #229 from SABS-R3-Epidemiology/kcgallagher-bugfix-…

…recovered-reinfection Bugfix - recovered reinfection
SABS-R3-Epidemiology · Sep 26, 2023 · 0afe665 · 0afe665
2 parents 86451d0 + db82fec
commit 0afe665
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyEpiabm/pyEpiabm/core/cell.py b/pyEpiabm/pyEpiabm/core/cell.py
@@ -82,15 +82,17 @@ def set_id(self, id: float):
         self.id = id
 
     def enqueue_person(self, person: Person):
-        """Add person to queue for processing at end of iteration.
+        """Add person to queue for processing at end of iteration, provided
+        they are not already recovered (and so may be infected).
 
         Parameters
         ----------
         person : Person
             Person to enqueue
 
         """
-        self.person_queue.put(person)
+        if person.infection_status != InfectionStatus.Recovered:
+            self.person_queue.put(person)
 
     def enqueue_PCR_testing(self, person: Person):
         """Add person to PCR testing queue for processing in testing
@@ -142,7 +144,7 @@ def notify_person_status_change(
 
     def number_infectious(self):
         """Returns the total number of infectious people in each
-        cell, all ages combined.
+         cell, all ages combined.
 
         Returns
         -------