From c3744da5d75d78441e2b8efaeeeb29060c472695 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Mon, 13 Jan 2020 11:50:12 +0100
Subject: [PATCH] only works for multivariate

---
 pts/model/tempflow/tempflow_estimator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pts/model/tempflow/tempflow_estimator.py b/pts/model/tempflow/tempflow_estimator.py
index ff577bb..cf6af57 100644
--- a/pts/model/tempflow/tempflow_estimator.py
+++ b/pts/model/tempflow/tempflow_estimator.py
@@ -110,13 +110,13 @@ class TempFlowEstimator(PTSEstimator):
             [
                 AsNumpyArray(
                     field=FieldName.TARGET,
-                    expected_ndim=1 + len(self.distr_output.event_shape),
+                    expected_ndim=2,
                 ),
                 # maps the target to (1, T)
                 # if the target data is uni dimensional
                 ExpandDimArray(
                     field=FieldName.TARGET,
-                    axis=0 if self.distr_output.event_shape[0] == 1 else None,
+                    axis=None,
                 ),
                 AddObservedValuesIndicator(
                     target_field=FieldName.TARGET,