relax type for generated y_t

CDCgov · Mar 6, 2024 · f80d941 · f80d941
1 parent 0c03621
commit f80d941
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/EpiAware/test/test_models.jl b/EpiAware/test/test_models.jl
@@ -32,7 +32,7 @@
     gen = generated_quantities(test_mdl, rand(test_mdl))
 
     #Check model sampled
-    @test eltype(gen.generated_y_t) <: Integer
+    @test eltype(gen.generated_y_t) <: AbstractFloat
     @test eltype(gen.I_t) <: AbstractFloat
     @test length(gen.I_t) == time_horizon
 end
@@ -71,7 +71,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Integer
+    @test eltype(gens[1].generated_y_t) <: AbstractFloat
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end
@@ -110,7 +110,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Integer
+    @test eltype(gens[1].generated_y_t) <: AbstractFloat
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end