fix tests

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

diff --git a/pkg/R/J_Neighbors.R b/pkg/R/J_Neighbors.R
index 7f66830..40341d9 100644 (file)
--- a/pkg/R/J_Neighbors.R
+++ b/pkg/R/J_Neighbors.R
@@ -17,11 +17,8 @@ getNeighborsJumpPredict = function(data, today, memory, horizon, params, ...)
        indices = params$indices[filter]
        weights = params$weights[filter]
 
-       if (any(is.na(weights) | is.na(indices)))
-               return (NA)
-
        gaps = sapply(indices, function(i) {
-               head( data$getSerie(i+1), 1) - tail( data$getSerie(i), 1)
+               head( data$getSerie(i+1),1 ) - tail( data$getSerie(i),1 )
        })
        scal_product = weights * gaps
        norm_fact = sum( weights[!is.na(scal_product)] )