Issue 511: Use benchmarkCI with DiffEqBase

.github/workflows/benchmark.yaml

@@ -22,6 +22,7 @@ jobs:
         run: |
           julia --project=benchmark -e 'using Pkg;
           Pkg.rm("EpiAware");
+          Pkg.add(url="https://github.com/SamuelBrand1/BenchmarkCI.jl");
           Pkg.resolve();
           Pkg.instantiate();
           Pkg.develop(path = "./EpiAware")'

benchmark/Project.toml

@@ -1,9 +1,10 @@
 [deps]
-BenchmarkCI = "20533458-34a3-403d-a444-e18f38190b5b"
 BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DynamicPPL = "366bfd00-2699-11ea-058f-f148b4cae6d8"
 EpiAware = "b2eeebe4-5992-4301-9193-7ebc9f62c855"
+OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 PkgBenchmark = "32113eaa-f34f-5b0d-bd6c-c81e245fc73d"
+SciMLSensitivity = "1ed8b502-d754-442c-8d5d-10ac956f44a1"
 Turing = "fce5fe82-541a-59a6-adf8-730c64b5f9a0"
 TuringBenchmarking = "0db1332d-5c25-4deb-809f-459bc696f94f"

benchmark/bench/EpiInfModels/EpiInfModels.jl

@@ -1,6 +1,7 @@
 module BenchEpiInfModels
 
-using BenchmarkTools, TuringBenchmarking, EpiAware, Distributions, OrdinaryDiffEq
+using BenchmarkTools, TuringBenchmarking, EpiAware, Distributions, OrdinaryDiffEq,
+      SciMLSensitivity
 suite = BenchmarkGroup()
 
 include("../../make_epiaware_suite.jl")

benchmark/bench/EpiInfModels/ODEProcess.jl

@@ -0,0 +1,20 @@
+let
+    # Create an instance of SIRParams
+    sirparams = SIRParams(
+        tspan = (0.0, 30.0),
+        infectiousness = LogNormal(log(0.3), 0.05),
+        recovery_rate = LogNormal(log(0.1), 0.05),
+        initial_prop_infected = Beta(1, 99)
+    )
+
+    # Define the ODEProcess
+    sir_process = ODEProcess(
+        params = sirparams,
+        sol2infs = sol -> sol[2, :],
+        solver_options = Dict(:verbose => false, :saveat => 1.0)
+    )
+
+    # Generate the latent infections
+    mdl = generate_latent_infs(sir_process, nothing)
+    suite["ODEProcess"] = make_epiaware_suite(mdl)
+end