fix bugs

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

diff --git a/pkg/R/Forecaster.R b/pkg/R/Forecaster.R
index a10f0bd..ef270b1 100644 (file)
--- a/pkg/R/Forecaster.R
+++ b/pkg/R/Forecaster.R
@@ -59,7 +59,7 @@ Forecaster = R6::R6Class("Forecaster",
                        # Parameters (potentially) computed during shape prediction stage
                        predicted_shape <- self$predictShape(data,today,memory,predict_from,horizon,...)
 
-                       if (is.na(predicted_shape))
+                       if (is.na(predicted_shape[1]))
                                return (NA)
 
                        predicted_delta <- private$.pjump(data, today, memory, predict_from,