Skip to content

Commit

Permalink
prior predictive pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
SamuelBrand1 committed Oct 6, 2024
1 parent af8c394 commit 482aebd
Showing 1 changed file with 13 additions and 4 deletions.
17 changes: 13 additions & 4 deletions pipeline/scripts/run_priorpred_pipeline.jl
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,11 @@ using Pkg
Pkg.activate(joinpath(@__DIR__(), ".."))
using Dagger

@assert !isempty(ARGS) "Test mode script requires the number of draws as an argument."
ndraws = parse(Int64, ARGS[1])

@info("""
Running the analysis pipeline.
Running the analysis pipeline in test mode with $(ndraws) draws per model.
--------------------------------------------
""")

Expand All @@ -15,11 +18,17 @@ pids = addprocs(; exeflags = ["--project=$(Base.active_project())"])

@everywhere using EpiAwarePipeline

# Create an instance of the pipeline behaviour
pipeline = RtwithoutRenewalPriorPipeline()
# Create instances of the pipeline behaviour

pipelines = [
SmoothOutbreakPipeline(ndraws = ndraws, nchains = 1, priorpredictive = true),
MeasuresOutbreakPipeline(ndraws = ndraws, nchains = 1, priorpredictive = true),
SmoothEndemicPipeline(ndraws = ndraws, nchains = 1, priorpredictive = true),
RoughEndemicPipeline(ndraws = ndraws, nchains = 1, priorpredictive = true),
]

# Run the pipeline
do_pipeline(pipeline)
do_pipeline(pipelines)

# Remove the workers
rmprocs(pids)

0 comments on commit 482aebd

Please sign in to comment.