diff --git a/examples/v1-client-tensorflow-mnist-classifier/demo.ipynb b/examples/v1-client-tensorflow-mnist-classifier/demo.ipynb index 134d00ba1..65b606d50 100644 --- 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)" ]