a few fixes

[talweg.git] / pkg / R / computeForecast.R

diff --git a/pkg/R/computeForecast.R b/pkg/R/computeForecast.R
index e967cc7..ca8badd 100644 (file)
--- a/pkg/R/computeForecast.R
+++ b/pkg/R/computeForecast.R
@@ -71,27 +71,20 @@ computeForecast = function(data, indices, forecaster, pjump, predict_from,
        forecaster = forecaster_class_name$new( #.pjump =
                getFromNamespace(paste("get",pjump,"JumpPredict",sep=""), "talweg"))
 
-       if (ncores > 1 && requireNamespace("parallel",quietly=TRUE))
+       computeOneForecast <- function(i)
        {
-               p <- parallel::mclapply(seq_along(integer_indices), function(i) {
-                       list(
-                               "forecast" = forecaster$predictSerie(
-                                       data, integer_indices[i], memory, predict_from, horizon, ...),
-                               "params"= forecaster$getParameters(),
-                               "index" = integer_indices[i] )
-                       }, mc.cores=ncores)
-       }
-       else
-       {
-               p <- lapply(seq_along(integer_indices), function(i) {
-                       list(
-                               "forecast" = forecaster$predictSerie(
-                                       data, integer_indices[i], memory, predict_from, horizon, ...),
-                               "params"= forecaster$getParameters(),
-                               "index" = integer_indices[i] )
-                       })
+               list(
+                       "forecast" = forecaster$predictSerie(data,i,memory,predict_from,horizon,...),
+                       "params" = forecaster$getParameters(),
+                       "index" = i )
        }
 
+       p <-
+               if (ncores > 1 && requireNamespace("parallel",quietly=TRUE))
+                       parallel::mclapply(integer_indices, computeOneForecast, mc.cores=ncores)
+               else
+                       lapply(integer_indices, computeOneForecast)
+
        # TODO: find a way to fill pred in //...
        for (i in seq_along(integer_indices))
        {