Changed LTSFLinearForecaster name to make it shorter.

djpasseyjr · Oct 2, 2024 · 28490be · 28490be
1 parent 5c492e8
commit 28490be
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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,