Skip to content

Commit

Permalink
change arg order
Browse files Browse the repository at this point in the history
  • Loading branch information
seabbs committed Mar 8, 2024
1 parent 47cdd14 commit ba47e50
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions EpiAware/src/latent-models.jl
Original file line number Diff line number Diff line change
Expand Up @@ -81,18 +81,18 @@ end

struct DiffLatentModel{T <: AbstractModel} <: AbstractLatentModel
model::T
d::Int
init_prior::Priors
d::Int

function DiffLatentModel(model::T, init_prior::Priors)
d = length(init_prior)
return new(model, d, init_prior)
return new(model, init_prior, d)
end

function DiffLatentModel(model::T, d::Int, init_prior::Priors)
@assert d>0 "d must be greater than 0"
@assert length(init_prior)==d "Length of init_prior must be equal to d"
return new(model, d, init_prior)
return new(model, init_prior, d)
end
end

Expand Down

0 comments on commit ba47e50

Please sign in to comment.