Merge branch 'main' into forecaster-train-simplify

ourownstory · Aug 30, 2024 · f3554be · f3554be
2 parents 7e0cc1f + 6807a3a
commit f3554be
Show file tree

Hide file tree

Showing 12 changed files with 891 additions and 907 deletions.
diff --git a/docs/source/how-to-guides/feature-guides/global_local_modeling_fut_regr.ipynb b/docs/source/how-to-guides/feature-guides/global_local_modeling_fut_regr.ipynb
@@ -750,7 +750,6 @@
    "source": [
     "try:\n",
     "    # it already installed dependencies\n",
-    "    from torchsummary import summary\n",
     "    from torchviz import make_dot\n",
     "except:\n",
     "    # install graphviz on system\n",
@@ -768,7 +767,6 @@
     "    !pip install torchviz\n",
     "    !pip install graphviz\n",
     "    # import\n",
-    "    from torchsummary import summary\n",
     "    from torchviz import make_dot"
    ]
   },

diff --git a/docs/source/how-to-guides/feature-guides/global_local_trend.ipynb b/docs/source/how-to-guides/feature-guides/global_local_trend.ipynb
@@ -1308,7 +1308,6 @@
    "source": [
     "try:\n",
     "    # it already installed dependencies\n",
-    "    from torchsummary import summary\n",
     "    from torchviz import make_dot\n",
     "except:\n",
     "    # install graphviz on system\n",
@@ -1326,7 +1325,6 @@
     "    !pip install torchviz\n",
     "    !pip install graphviz\n",
     "    # import\n",
-    "    from torchsummary import summary\n",
     "    from torchviz import make_dot"
    ]
   },

diff --git a/docs/source/how-to-guides/feature-guides/mlflow.ipynb b/docs/source/how-to-guides/feature-guides/mlflow.ipynb
@@ -237,6 +237,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "7fb27b941602401d91542211134fc71a",
    "metadata": {},
    "source": [
     "## Advanced Example \n",
@@ -246,6 +247,7 @@
   {
    "cell_type": "code",
    "execution_count": 7,
+   "id": "acae54e37e7d407bbb7b55eff062a284",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -254,7 +256,6 @@
     "# Copy and paste url from command line to web browser\n",
     "\n",
     "import mlflow\n",
-    "import torchmetrics\n",
     "from mlflow.data.pandas_dataset import PandasDataset\n",
     "\n",
     "if local:\n",
@@ -272,7 +273,6 @@
     "    )\n",
     "\n",
     "    import mlflow.pytorch\n",
-    "    from mlflow.client import MlflowClient\n",
     "\n",
     "    model_name = \"NeuralProphet\"\n",
     "\n",