[FIX] Missing autormok in core.ipynb (#1167)

Nixtla · Oct 1, 2024 · 38195f6 · 38195f6
1 parent 69255e8
commit 38195f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -93,7 +93,7 @@
     "    StemGNN, PatchTST, TimesNet, TimeLLM, TSMixer, TSMixerx,\n",
     "    MLPMultivariate, iTransformer,\n",
     "    BiTCN, TiDE, DeepNPTS, SOFTS,\n",
-    "    TimeMixer, KAN\n",
+    "    TimeMixer, KAN, RMoK\n",
     ")\n",
     "from neuralforecast.common._base_auto import BaseAuto, MockTrial"
    ]
@@ -245,7 +245,8 @@
     "    'deepnpts': DeepNPTS, 'autodeepnpts': DeepNPTS,\n",
     "    'softs': SOFTS, 'autosofts': SOFTS,\n",
     "    'timemixer': TimeMixer, 'autotimemixer': TimeMixer,\n",
-    "    'kan': KAN, 'autokan': KAN\n",
+    "    'kan': KAN, 'autokan': KAN,\n",
+    "    'rmok': RMoK, 'autormok': RMoK\n",
     "}"
    ]
   },

diff --git a/neuralforecast/core.py b/neuralforecast/core.py
@@ -66,6 +66,7 @@
     SOFTS,
     TimeMixer,
     KAN,
+    RMoK,
 )
 from .common._base_auto import BaseAuto, MockTrial
 
@@ -190,6 +191,8 @@ def _insample_times(
     "autotimemixer": TimeMixer,
     "kan": KAN,
     "autokan": KAN,
+    "rmok": RMoK,
+    "autormok": RMoK,
 }
 
 # %% ../nbs/core.ipynb 8