From 28490be181095b1f08a4ec64f303b5b9a34f6491 Mon Sep 17 00:00:00 2001 From: DJ Passey Date: Wed, 2 Oct 2024 16:03:36 -0400 Subject: [PATCH] Changed LTSFLinearForecaster name to make it shorter. --- interfere/_methods/deep_learning.py | 4 ++-- interfere/methods/__init__.py | 2 +- tests/test_methods.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/interfere/_methods/deep_learning.py b/interfere/_methods/deep_learning.py index 2453fd7..e9ad5e6 100644 --- a/interfere/_methods/deep_learning.py +++ b/interfere/_methods/deep_learning.py @@ -9,7 +9,7 @@ raise ImportError( "ImportError occured in sktime import." "\n\n This likely occurred because `interfere` does not list `sktime` as a direct dependency. To use the " - "LTSFLinearForecaster, try first install `sktime` via pip " + "LTSF, try first install `sktime` via pip " "install sktime." f"\n\nOriginating error text: {e}" ) @@ -18,7 +18,7 @@ from ..utils import copy_doc, to_sktime_time_series -class LTSFLinearForecaster(ForecastMethod): +class LTSF(ForecastMethod): """Uses a transformer for inference.""" diff --git a/interfere/methods/__init__.py b/interfere/methods/__init__.py index b0252ce..010ab96 100644 --- a/interfere/methods/__init__.py +++ b/interfere/methods/__init__.py @@ -11,7 +11,7 @@ """ from .._methods.average_method import AverageMethod -from .._methods.deep_learning import LTSFLinearForecaster +from .._methods.deep_learning import LTSF from .._methods.nixtla_methods.neuralforecast_methods import LSTM, NHITS from .._methods.nixtla_methods.statsforecast_methods import AutoARIMA from .._methods.reservoir_computer import ResComp diff --git a/tests/test_methods.py b/tests/test_methods.py index 985a2c2..1b6682a 100644 --- a/tests/test_methods.py +++ b/tests/test_methods.py @@ -22,7 +22,7 @@ METHODS = [ interfere.methods.ResComp, interfere.methods.SINDY, - interfere.methods.LTSFLinearForecaster, + interfere.methods.LTSF, interfere.methods.VAR, interfere.methods.LSTM, interfere.methods.NHITS,