Skip to content

Commit

Permalink
Fix ExponentialSmoothing.
Browse files Browse the repository at this point in the history
  • Loading branch information
csadorf committed Feb 6, 2023
1 parent 40ec3f4 commit 440d6ab
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/cuml/tsa/holtwinters.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ np = cpu_only_import('numpy')
from libc.stdint cimport uintptr_t

import cuml.internals
from cuml.internals.input_utils import input_to_cupy_array
from cuml.internals.input_utils import input_to_cuml_array, input_to_cupy_array
from cuml.internals import _deprecate_pos_args
from cuml.common import using_output_type
from cuml.internals.base import Base
Expand Down Expand Up @@ -244,6 +244,7 @@ class ExponentialSmoothing(Base):
# Set up attributes:
self.eps = eps
self.endog = endog
self._set_output_type(endog)
self.forecasted_points = [] # list for final forecast output
self.level = [] # list for level values for each time series in batch
self.trend = [] # list for trend values for each time series in batch
Expand Down Expand Up @@ -279,7 +280,7 @@ class ExponentialSmoothing(Base):
raise ValueError(err_mess + str(d2))
else:
raise ValueError("Data input must have 1 or 2 dimensions.")
return mod_ts_input
return input_to_cuml_array(mod_ts_input).array

@cuml.internals.api_base_return_any_skipall
def fit(self) -> "ExponentialSmoothing":
Expand Down

0 comments on commit 440d6ab

Please sign in to comment.