From 715772bbcc355b629b7579fcaed1f602761c41a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido?= <petronio.candido@gmail.com>
Date: Mon, 27 Jan 2020 19:30:26 -0300
Subject: [PATCH] Removing direct pyflux import from benchmark.BSTS modules

---
 pyFTS/benchmarks/BSTS.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyFTS/benchmarks/BSTS.py b/pyFTS/benchmarks/BSTS.py
index 11c8838..bd4f9cc 100644
--- a/pyFTS/benchmarks/BSTS.py
+++ b/pyFTS/benchmarks/BSTS.py
@@ -3,7 +3,6 @@
 
 import numpy as np
 import pandas as pd
-import pyflux as pf
 import scipy.stats as st
 from pyFTS.common import SortedCollection, fts
 from pyFTS.probabilistic import ProbabilityDistribution
@@ -46,6 +45,7 @@ class ARIMA(fts.FTS):
             self.shortname = "BSTS({},{},{})-{}".format(self.p,self.d,self.q,self.alpha)
 
     def train(self, data, **kwargs):
+        import pyflux as pf
 
         if 'order' in kwargs:
             order = kwargs.pop('order')
@@ -95,8 +95,9 @@ class ARIMA(fts.FTS):
         return ret
 
     def forecast_distribution(self, data, **kwargs):
+        import pyflux as pf
 
-        sim_vector = self.inference(steps)
+        sim_vector = self.inference(1)
 
         ret = []