Materilaizers loaded in nb

zenml-io · Aug 22, 2024 · 678034a · 678034a
1 parent c2a918d
commit 678034a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/quickstart/quickstart.ipynb b/examples/quickstart/quickstart.ipynb
@@ -173,6 +173,7 @@
     "from typing_extensions import Annotated\n",
     "\n",
     "from zenml import step\n",
+    "import materializers\n",
     "\n",
     "PROMPT = \"\"  # In case you want to also use a prompt you can set it here\n",
     "\n",
@@ -251,6 +252,7 @@
     "\n",
     "from steps import load_data, tokenize_data, train_model, evaluate_model, test_model, split_dataset\n",
     "from steps.model_trainer import T5_Model\n",
+    "import materializers\n",
     "\n",
     "# Initialize the ZenML client to fetch objects from the ZenML Server\n",
     "client = Client()\n",
@@ -437,6 +439,7 @@
     "from zenml.config import DockerSettings, ResourceSettings\n",
     "\n",
     "from steps.model_trainer import T5_Model\n",
+    "import materializers\n",
     "\n",
     "settings={\n",
     "    \"docker\": DockerSettings(\n",