diff --git a/etna/models/base.py b/etna/models/base.py index 7fd7a69e0..dccd4ca3f 100644 --- a/etna/models/base.py +++ b/etna/models/base.py @@ -632,8 +632,7 @@ def forecast(self, ts: "TSDataset", prediction_size: int) -> "TSDataset": future_ts = ts.tsdataset_idx_slice(start_idx=self.encoder_length, end_idx=self.encoder_length + prediction_size) for (segment, feature_nm), value in predictions.items(): # we don't want to change dtype after assignment, but there can happen cast to float32 - dtype = future_ts.df.loc[:, pd.IndexSlice[segment, feature_nm]] - future_ts.df.loc[:, pd.IndexSlice[segment, feature_nm]] = value[:prediction_size, :].astype(dtype) + future_ts.df.loc[:, pd.IndexSlice[segment, feature_nm]] = value[:prediction_size, :].astype(np.float64) future_ts.inverse_transform()