From 0fdebe2f60dc066376a76b1480164f5b4a24375b Mon Sep 17 00:00:00 2001
From: Guido Espana <ukd0@cdc.gov>
Date: Thu, 19 Dec 2024 13:39:02 +0000
Subject: [PATCH] Use sample_person with query

---
 src/transmission_manager.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/transmission_manager.rs b/src/transmission_manager.rs
index 523404e..d0d0629 100644
--- a/src/transmission_manager.rs
+++ b/src/transmission_manager.rs
@@ -7,6 +7,7 @@ use statrs::distribution::{ContinuousCDF, Exp, Poisson};
 
 use crate::contact::ContextContactExt;
 use crate::parameters::Parameters;
+use crate::population_loader::Alive;
 
 // Define the possible infectious statuses for a person.
 // These states refer to the person's infectiousness at a given time
@@ -46,7 +47,7 @@ pub fn init(context: &mut Context) {
 fn seed_infections(context: &mut Context) -> Result<(), IxaError> {
     // For now, we just pick a random person and make them infectious.
     // In the future, we may pick people based on specific person properties.
-    let random_person_id = context.sample_person(TransmissionRng)?;
+    let random_person_id = context.sample_person(TransmissionRng, (Alive, true))?;
     context.set_person_property(
         random_person_id,
         InfectiousStatus,