update path

Signed-off-by: Wesley M. Gifford <[email protected]>
ibm-granite · Apr 8, 2024 · b67d68c · b67d68c
1 parent 85b7d68
commit b67d68c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/hfdemo/ttm_getting_started.ipynb b/notebooks/hfdemo/ttm_getting_started.ipynb
@@ -40,14 +40,14 @@
     "\n",
     "# First Party\n",
     "from notebooks.hfdemo.tinytimemixer.utils import (\n",
-    "    TrackingCallback,\n",
     "    count_parameters,\n",
     "    get_data,\n",
     "    plot_preds,\n",
     ")\n",
     "\n",
     "# Local\n",
-    "from tsfm_public.models.tinytimemixer import TinyTimeMixerForPrediction"
+    "from tsfm_public.models.tinytimemixer import TinyTimeMixerForPrediction\n",
+    "from tsfm_public.toolit.callbacks import TrackingCallback"
    ]
   },
   {