autogluon.timeseries 0.8.3b20231004__py3-none-any.whl → 0.8.3b20231006__py3-none-any.whl

autogluon/timeseries/evaluator.py

@@ -2,7 +2,6 @@
 See also, https://ts.gluon.ai/api/gluonts/gluonts.evaluation.html
 """
 import logging
-import warnings
 from typing import Optional
 
 import numpy as np
@@ -11,7 +10,7 @@ import pandas as pd
 from autogluon.timeseries import TimeSeriesDataFrame
 from autogluon.timeseries.dataset.ts_dataframe import ITEMID
 from autogluon.timeseries.utils.seasonality import get_seasonality
-from autogluon.timeseries.utils.warning_filters import evaluator_warning_filter
+from autogluon.timeseries.utils.warning_filters import warning_filter
 
 logger = logging.getLogger(__name__)
 
@@ -268,10 +267,7 @@ class TimeSeriesEvaluator:
 
         assert data_future.index.equals(predictions.index), "Prediction and data indices do not match."
 
-        with evaluator_warning_filter(), warnings.catch_warnings():
-            warnings.simplefilter("ignore", category=UserWarning)
-            warnings.simplefilter("ignore", category=RuntimeWarning)
-            warnings.simplefilter("ignore", category=FutureWarning)
+        with warning_filter():
             return self.metric_method(
                 y_true=data_future[self.target_column],
                 predictions=predictions,

autogluon/timeseries/models/autogluon_tabular/__init__.py

@@ -1,5 +1,4 @@
-from .direct_tabular import DirectTabularModel
-from .mlforecast import RecursiveTabularModel
+from .mlforecast import DirectTabularModel, RecursiveTabularModel
 
 __all__ = [
     "DirectTabularModel",