Fix JuliaBackend calibrate constructor

CliMA · Oct 18, 2024 · dd12d6e · dd12d6e
1 parent 5dd782a
commit dd12d6e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "ClimaCalibrate"
 uuid = "4347a170-ebd6-470c-89d3-5c705c0cacc2"
 authors = ["Climate Modeling Alliance"]
-version = "0.0.4"
+version = "0.0.5"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
@@ -22,7 +22,7 @@ CESExt = "CalibrateEmulateSample"
 CalibrateEmulateSample = "0.5"
 Distributed = "1"
 Distributions = "0.25"
-EnsembleKalmanProcesses = "1, 2"
+EnsembleKalmanProcesses = "2"
 JLD2 = "0.4, 0.5"
 Random = "1"
 TOML = "1"

diff --git a/src/backends.jl b/src/backends.jl
@@ -82,8 +82,12 @@ function calibrate(
     ::Type{JuliaBackend},
     config::ExperimentConfig;
     reruns = 0,
+    model_interface = nothing,
     ekp_kwargs...,
 )
+    if !isnothing(model_interface)
+        @everywhere include(model_interface)
+    end
     (; n_iterations, output_dir, ensemble_size) = config
     eki = initialize(config; ekp_kwargs...)
     on_error(e::InterruptException) = rethrow(e)

diff --git a/src/ekp_interface.jl b/src/ekp_interface.jl
@@ -269,7 +269,7 @@ function _initialize(
         EKP.construct_initial_ensemble(rng_ekp, prior, ensemble_size)
 
     eki_constructor =
-        (args...) -> EKP.EnsembleKalmanProcess(
+         (args...) -> EKP.EnsembleKalmanProcess(
             args...,
             Dict(EKP.default_options_dict(EKP.Inversion())..., ekp_kwargs...);
             rng = rng_ekp,