From 797e893edafe7f42ad1c705fbe0291c0c45e8705 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sun, 23 May 2021 23:14:10 +0200
Subject: [PATCH] Print experiment info

---
 mlflow_prediction_registry.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py
index 25b700e..bcee87b 100644
--- a/mlflow_prediction_registry.py
+++ b/mlflow_prediction_registry.py
@@ -18,9 +18,9 @@ model_name = "s426206"
 
 input = str(arguments[0])
 
-model = mlflow.pyfunc.load_model(
-    model_uri=f"models:/{model_name}/{model_version}"
-)
+experiment = client.get_latest_versions(model_name, stages=None)
+
+print(experiment)
 
 with open(f'{model_name}/{input}', 'r') as file:
     json_data = json.load(file)