examples: add uuid's to demo

usnistgov · Sep 10, 2024 · 37c7d37 · 37c7d37
1 parent a0fa394
commit 37c7d37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/v1-client-tensorflow-mnist-classifier/demo.ipynb b/examples/v1-client-tensorflow-mnist-classifier/demo.ipynb
@@ -627,9 +627,9 @@
    "outputs": [],
    "source": [
     "from mlflow.tracking import MlflowClient\n",
+    "from uuid import UUID\n",
     "mlflow_client = MlflowClient()\n",
-    "mlflow_runid = client.jobs.get_mlflow_run_id(infer_job['id'])['mlflowRunId'].replace('-','') # why\n",
-    "print(mlflow_runid)\n",
+    "mlflow_runid = UUID(client.jobs.get_mlflow_run_id(infer_job['id'])['mlflowRunId']).hex\n",
     "mlflow_run = mlflow_client.get_run(mlflow_runid)\n",
     "pprint.pprint(mlflow_run.data.metrics)"
    ]