Merge pull request #533 from GoogleCloudPlatform/kfp_wt_fix

KFP walkthrough minor fix
GoogleCloudPlatform · Dec 12, 2024 · d5f3728 · d5f3728
2 parents 83b0a1a + 0d222c6
commit d5f3728
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/kubeflow_pipelines/walkthrough/solutions/kfp_walkthrough_vertex.ipynb b/notebooks/kubeflow_pipelines/walkthrough/solutions/kfp_walkthrough_vertex.ipynb
@@ -632,7 +632,7 @@
    "source": [
     "def get_trials(job_name):\n",
     "    jobs = aiplatform.HyperparameterTuningJob.list()\n",
-    "    match = [job for job in jobs if job.display_name == JOB_NAME]\n",
+    "    match = [job for job in jobs if job.display_name == job_name]\n",
     "    tuning_job = match[0] if match else None\n",
     "    return tuning_job.trials if tuning_job else None\n",
     "\n",