Fixed issues:

- Inverse Box-Cox calculation issue #1
intive-DataScience · Jan 18, 2019 · a499501 · a499501
1 parent 51a8686
commit a499501
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -43,7 +43,7 @@ def run_tests(self):
 
 setuptools.setup(
     name='tbats',
-    version='1.0.4',
+    version='1.0.5',
     packages=setuptools.find_packages(exclude=('test', 'test_R')),
     url='https://github.com/intive-DataScience/tbats',
     license='MIT License',

diff --git a/tbats/abstract/Model.py b/tbats/abstract/Model.py
@@ -226,7 +226,13 @@ def _fit_to_observations(self, y, starting_x):
         # store fit results
         self.x_last = x
         self.resid_boxcox = yw - yw_hat
-        self.y_hat = self._inv_boxcox(yw_hat)
+        try:
+            self.y_hat = self._inv_boxcox(yw_hat)
+        except RuntimeWarning:
+            self.add_warning('Box-Cox related numeric calculation issues detected. Model is not usable.')
+            self.is_fitted = False
+            return self
+
         self.resid = self.y - self.y_hat
 
         self.is_fitted = True