add some more convenience functions for ObservationSeries

CliMA · Jun 21, 2024 · 291f59c · 291f59c
1 parent 5e0c301
commit 291f59c
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/EnsembleKalmanProcess.jl b/src/EnsembleKalmanProcess.jl
@@ -507,6 +507,17 @@ function get_obs(ekp::EnsembleKalmanProcess; build = true)
     return get_obs(get_observation_series(ekp), build = build)
 end
 
+"""
+    update_minibatch!(ekp::EnsembleKalmanProcess)
+update to the next minibatch in the ObservationSeries
+"""
+function update_minibatch!(ekp::EnsembleKalmanProcess)
+    return update_minibatch!(get_observation_series(ekp))
+end
+
+function get_current_minibatch(ekp::EnsembleKalmanProcess)
+    return get_current_minibatch(get_observation_series(ekp))
+end
 
 """
     construct_initial_ensemble(
@@ -736,8 +747,8 @@ function update_ensemble!(
     end
 
     # update to next minibatch (if minibatching)
-    observation_series = get_observation_series(ekp)
-    next_minibatch = update_minibatch!(observation_series)
+    next_minibatch = update_minibatch!(ekp)
+
     return nothing
 
 end