From 5d8fba3486aa2739b154162dfe5b9921cf8c8058 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sun, 23 May 2021 23:38:54 +0200
Subject: [PATCH] Print experiment info

---
 mlflow_prediction_registry.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py
index f8c5736..4507887 100644
--- a/mlflow_prediction_registry.py
+++ b/mlflow_prediction_registry.py
@@ -22,7 +22,7 @@ experiment = client.get_latest_versions(model_name, stages=None)
 
 print(experiment)
 
-with open(f'{experiment.source}/{input}', 'r') as file:
+with open(f'{experiment[0].source}/{input}', 'r') as file:
     json_data = json.load(file)
 
 print(model(torch.tensor(np.array(json_data['inputs'])).float()))
\ No newline at end of file