Skip to content

Commit

Permalink
return filterwarning to script
Browse files Browse the repository at this point in the history
  • Loading branch information
Egor Baturin committed Jan 31, 2025
1 parent d9a7d27 commit a84122a
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions etna/models/sarimax.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@

import numpy as np
import pandas as pd
from statsmodels.tools.sm_exceptions import ValueWarning
from statsmodels.tsa.statespace.sarimax import SARIMAX
from statsmodels.tsa.statespace.sarimax import SARIMAXResultsWrapper
from statsmodels.tsa.statespace.simulation_smoother import SimulationSmoother
Expand All @@ -25,6 +26,13 @@
from etna.models.mixins import PredictionIntervalContextIgnorantModelMixin
from etna.models.utils import select_observations

warnings.filterwarnings(
message="No frequency information was provided, so inferred frequency .* will be used",
action="ignore",
category=ValueWarning,
module="statsmodels.tsa.base.tsa_model",
)

_DEFAULT_FREQ = object()


Expand Down

0 comments on commit a84122a

Please sign in to comment.