From d54197cd6a7494d867948a79612905f0f8d8aafc Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Mon, 17 Feb 2020 14:39:04 +0100
Subject: [PATCH] fix prediction length

---
 pts/model/forecast.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pts/model/forecast.py b/pts/model/forecast.py
index 2a329be..1ca5dc6 100644
--- a/pts/model/forecast.py
+++ b/pts/model/forecast.py
@@ -249,7 +249,7 @@ class SampleForecast(Forecast):
         """
         Time length of the forecast.
         """
-        return self.samples.shape[-1]
+        return self.samples.shape[1]
 
     @property
     def mean(self):