Fix the truthdata output to be missing or Int (#429)

CDCgov · Aug 16, 2024 · aca23b1 · aca23b1
1 parent 46344f2
commit aca23b1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pipeline/src/simulate/TruthSimulationConfig.jl b/pipeline/src/simulate/TruthSimulationConfig.jl
@@ -67,7 +67,7 @@ function simulate(config::TruthSimulationConfig)
         (std = 1.0, cluster_factor = config.cluster_factor,
             ϵ_t = config.logit_daily_ascertainment))
 
-    y_t = obs_model()
+    y_t = obs_model() .|> y -> ismissing(y) ? missing : Int(y)
 
     #Return the sampled infections and observations
 

diff --git a/pipeline/test/pipeline/test_pipelinefunctions.jl b/pipeline/test/pipeline/test_pipelinefunctions.jl
@@ -5,7 +5,7 @@
     truthdata = fetch.(truthdata_dg_task)
 
     @test length(truthdata) == 1
-    @test all([data["y_t"] isa Vector{Union{Missing, T}} where {T <: Real}
+    @test all([data["y_t"] isa Vector{Union{Missing, T}} where {T <: Integer}
                for data in truthdata])
 end