More imports

zenml-io · Aug 22, 2024 · 515fd61 · 515fd61
1 parent 678034a
commit 515fd61
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/quickstart/quickstart.ipynb b/examples/quickstart/quickstart.ipynb
@@ -439,6 +439,8 @@
     "from zenml.config import DockerSettings, ResourceSettings\n",
     "\n",
     "from steps.model_trainer import T5_Model\n",
+    "from steps import load_data, tokenize_data, train_model, evaluate_model, test_model, split_dataset\n",
+    "\n",
     "import materializers\n",
     "\n",
     "settings={\n",
@@ -453,7 +455,7 @@
     "    !pip install 'kfp>=2.6.0' gcsfs google-cloud-secret-manager 'google-cloud-container>=2.21.0' 'google-cloud-artifact-registry>=1.11.3' 'google-cloud-storage>=2.9.0' 'google-cloud-aiplatform>=1.34.0' 'google-cloud-build>=3.11.0' kubernetes\n",
     "    \n",
     "    # Set the name of your gcp stack here\n",
-    "    stack_name = None # INSERT_STACK_NAME_HERE\n",
+    "    stack_name = \"alexej-gcp-quickstart-stack2\" # INSERT_STACK_NAME_HERE\n",
     "    \n",
     "elif CLOUD_PROVIDER == \"AWS\":\n",
     "    !pip install 'sagemaker>=2.117.0' s3 s3fs aws-profile-manager kubernetes\n",