add dependencies to the dev

Nixtla · Nov 25, 2024 · 87ab801 · 87ab801
1 parent 1d762bd
commit 87ab801
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/nbs/docs/getting-started/7_why_timegpt.ipynb b/nbs/docs/getting-started/7_why_timegpt.ipynb
@@ -1,15 +1,5 @@
 {
  "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#| hide\n",
-    "!pip install -Uqq nixtla statsforecast mlforecast neuralforecast lightgbm"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -393,7 +383,7 @@
      "data": {
       "text/plain": [
        "metric\n",
-       "rmse     592.588597\n",
+       "rmse     592.603154\n",
        "smape      0.049403\n",
        "Name: TimeGPT, dtype: float64"
       ]
@@ -450,7 +440,7 @@
    "outputs": [],
    "source": [
     "#| eval: false\n",
-    "Initiate ARIMA model\n",
+    "#Initiate ARIMA model\n",
     "sf = StatsForecast(\n",
     "    models=[AutoARIMA(season_length=7)],\n",
     "    freq='D'\n",
@@ -620,7 +610,7 @@
    "outputs": [],
    "source": [
     "#| eval: false\n",
-    "Initialize the N-HiTS model.\n",
+    "# Initialize the N-HiTS model.\n",
     "models = [NHITS(h=28, \n",
     "                input_size=28, \n",
     "                max_steps=100)]\n",

diff --git a/setup.py b/setup.py
@@ -21,6 +21,9 @@
     "setuptools<70",
     "statsforecast",
     "tabulate",
+    "nixtla",
+    "mlforecast",
+    "lightgbm"
 ]
 distributed = ["fugue[dask,ray,spark]>=0.8.7", "pandas<2.2", "ray<2.6.3"]
 plotting = ["utilsforecast[plotting]>=0.2.7"]