From c4f264a1f263e8aec1ec5895b800f98ec50d541f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido=20de=20Lima=20e=20Silva?=
 <petronio.candido@gmail.com>
Date: Thu, 28 Jun 2018 14:31:30 -0300
Subject: [PATCH] Bugfix in Measures.get_point_statistics for multivariate
 models

---
 pyFTS/benchmarks/Measures.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyFTS/benchmarks/Measures.py b/pyFTS/benchmarks/Measures.py
index 6aadfc5..b07fd47 100644
--- a/pyFTS/benchmarks/Measures.py
+++ b/pyFTS/benchmarks/Measures.py
@@ -323,6 +323,9 @@ def get_point_statistics(data, model, **kwargs):
 
     if steps_ahead == 1:
         forecasts = model.predict(ndata, **kwargs)
+        
+        if model.is_multivariate:
+            ndata = ndata[model1.target_variable.data_label].values
 
         if not isinstance(forecasts, (list, np.ndarray)):
             forecasts = [forecasts]