From 7ea23317004fdbaac5d6fdad3f716827263adc8a Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Sun, 12 Jan 2020 13:45:30 +0100
Subject: [PATCH] fix sample() method

---
 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 9a5a5ed..ef3fbf0 100644
--- a/pts/model/forecast.py
+++ b/pts/model/forecast.py
@@ -530,7 +530,7 @@ class DistributionForecast(Forecast):
 
     def to_sample_forecast(self, num_samples: int = 200) -> SampleForecast:
         return SampleForecast(
-            samples=self.distribution.sample(num_samples),
+            samples=self.distribution.sample((num_samples,)),
             start_date=self.start_date,
             freq=self.freq,
             item_id=self.item_id,