Fixing the quickstart bug

zenml-io · Feb 6, 2024 · 99a1a08 · 99a1a08
1 parent ef7cbb1
commit 99a1a08
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/template/quickstart.ipynb b/template/quickstart.ipynb
@@ -628,8 +628,8 @@
     "        dataset_trn, dataset_tst = feature_engineering()\n",
     "    else:\n",
     "        # Load the datasets from an older pipeline\n",
-    "        dataset_trn = client.get_artifact_version(id=train_dataset_id)\n",
-    "        dataset_tst = client.get_artifact_version(id=test_dataset_id) \n",
+    "        dataset_trn = client.get_artifact_version(name_id_or_prefix=train_dataset_id)\n",
+    "        dataset_tst = client.get_artifact_version(name_id_or_prefix=test_dataset_id) \n",
     "\n",
     "    trained_model = model_trainer(\n",
     "        dataset_trn=dataset_trn,\n",
@@ -970,8 +970,8 @@
     "@pipeline\n",
     "def inference(preprocess_pipeline_id: UUID):\n",
     "    \"\"\"Model batch inference pipeline\"\"\"\n",
-    "    # random_state = client.get_artifact_version(id=preprocess_pipeline_id).metadata[\"random_state\"].value\n",
-    "    # target = client.get_artifact_version(id=preprocess_pipeline_id).run_metadata['target'].value\n",
+    "    # random_state = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).metadata[\"random_state\"].value\n",
+    "    # target = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).run_metadata['target'].value\n",
     "    random_state = 42\n",
     "    target = \"target\"\n",
     "\n",
@@ -981,7 +981,7 @@
     "    df_inference = inference_preprocessor(\n",
     "        dataset_inf=df_inference,\n",
     "        # We use the preprocess pipeline from the feature engineering pipeline\n",
-    "        preprocess_pipeline=client.get_artifact_version(id=preprocess_pipeline_id),\n",
+    "        preprocess_pipeline=client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id),\n",
     "        target=target,\n",
     "    )\n",
     "    inference_predict(\n",