From bda0aec524f66b4b5e3a6b1ca5bef2e8e6e41ac2 Mon Sep 17 00:00:00 2001
From: Filip Izydorczyk <filip.izydorczyk@protonmail.com>
Date: Thu, 10 Jun 2021 14:45:52 +0200
Subject: [PATCH] Testing MlflowClient

---
 predict-registry/predict.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/predict-registry/predict.py b/predict-registry/predict.py
index dc3daa3..22e22c5 100644
--- a/predict-registry/predict.py
+++ b/predict-registry/predict.py
@@ -12,3 +12,6 @@ mlflow.set_tracking_uri("http://172.17.0.1:5000")
 model = mlflow.keras.pyfunc.load_model(
     model_uri=f"models:/{model_name}/{model_version}",
 )
+
+client = MlflowClient()
+print(client.get_model_version(model_name, model_version).source)