Testing MlflowClient

2021-06-10 14:45:52 +02:00 · 2021-06-10 14:45:52 +02:00 · bda0aec524
commit bda0aec524
parent 805da075ae
1 changed files with 3 additions and 0 deletions
--- 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)