correct type specification

CDCgov · Jun 11, 2024 · d5f847d · d5f847d
1 parent ea2b6df
commit d5f847d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/EpiAware/src/EpiObsModels/ObservationErrorModels/NegativeBinomialError.jl b/EpiAware/src/EpiObsModels/ObservationErrorModels/NegativeBinomialError.jl
@@ -14,7 +14,7 @@ nb_model = generate_observations(nb, missing, fill(10, 10))
 rand(nb_model)
 ```
 "
-@kwdef struct NegativeBinomialError{S <: Sampleable, T <: AbstractFloat} <:
+@kwdef struct NegativeBinomialError{S <: Sampleable} <:
               AbstractTuringObservationErrorModel
     "The prior distribution for the cluster factor."
     cluster_factor_prior::S = HalfNormal(0.01)

diff --git a/EpiAware/src/EpiObsModels/ObservationErrorModels/PoissonError.jl b/EpiAware/src/EpiObsModels/ObservationErrorModels/PoissonError.jl
@@ -13,7 +13,7 @@ poi_model = generate_observations(poi, missing, fill(10, 10))
 rand(poi_model)
 ```
 "
-struct PoissonError{T <: AbstractFloat} <: AbstractTuringObservationErrorModel
+struct PoissonError <: AbstractTuringObservationErrorModel
 end
 
 @doc raw"

diff --git a/EpiAware/src/EpiObsModels/ObservationErrorModels/methods.jl b/EpiAware/src/EpiObsModels/ObservationErrorModels/methods.jl
@@ -15,7 +15,7 @@ It dispatches to the `observation_error` function to generate the observation er
         @assert length(y_t)==length(Y_t) "The observation vector and expected observation vector must have the same length."
     end
 
-    pad_Y_t = Y_t + 1e-6
+    pad_Y_t = Y_t .+ 1e-6
 
     for i in findfirst(!ismissing, Y_t):length(Y_t)
         y_t[i] ~ observation_error(obs_model, pad_Y_t[i], priors...)