Latest changes

zenml-io · Dec 21, 2023 · 56360c9 · 56360c9
1 parent a4c1682
commit 56360c9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/template/quickstart.ipynb b/template/quickstart.ipynb
@@ -568,7 +568,7 @@
     "def model_trainer(\n",
     "    dataset_trn: pd.DataFrame,\n",
     "    model_type: str = \"sgd\",\n",
-    ") -> Annotated[ClassifierMixin, ArtifactConfig(name=\"model\", is_model_artifact=True)]:\n",
+    ") -> Annotated[ClassifierMixin, ArtifactConfig(name=\"sklearn_classifier\", is_model_artifact=True)]:\n",
     "    \"\"\"Configure and train a model on the training dataset.\"\"\"\n",
     "    target = \"target\"\n",
     "    if model_type == \"sgd\":\n",
@@ -840,7 +840,7 @@
     "rf_zenml_model_version = client.get_model_version(\"breast_cancer_classifier\", \"rf\")\n",
     "\n",
     "# We can now load our classifier directly as well\n",
-    "random_forest_classifier = rf_zenml_model_version.get_artifact(\"model\").load()\n",
+    "random_forest_classifier = rf_zenml_model_version.get_artifact(\"sklearn_classifier\").load()\n",
     "\n",
     "random_forest_classifier"
    ]
@@ -943,7 +943,7 @@
     "    model_version = get_step_context().model_version\n",
     "\n",
     "    # run prediction from memory\n",
-    "    predictor = model_version.load_artifact(\"model\")\n",
+    "    predictor = model_version.load_artifact(\"sklearn_classifier\")\n",
     "    predictions = predictor.predict(dataset_inf)\n",
     "\n",
     "    predictions = pd.Series(predictions, name=\"predicted\")\n",
@@ -1114,7 +1114,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.13"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,