From a3b9ff846da103fb9ae956e4a0f530b819792488 Mon Sep 17 00:00:00 2001
From: varkha-d-sharma <112053040+varkha-d-sharma@users.noreply.github.com>
Date: Tue, 6 Feb 2024 08:07:09 +0530
Subject: [PATCH] Fixed GUI bug (#143)

* server/app/get_data.py

* changed logic for artifact retrieval
---
 server/app/get_data.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/server/app/get_data.py b/server/app/get_data.py
index 5e01bb82..eaaf2a83 100644
--- a/server/app/get_data.py
+++ b/server/app/get_data.py
@@ -39,10 +39,15 @@ def get_all_artifact_ids(mlmdfilepath):
     artifact_ids = {}
     query = cmfquery.CmfQuery(mlmdfilepath)
     names = query.get_pipeline_names()
+    execution_ids = get_all_exe_ids(mlmdfilepath)
     for name in names:
         df = pd.DataFrame()
-        artifacts = query.get_all_artifacts_by_context(name)
-        df = pd.concat([df, artifacts], sort=True, ignore_index=True)
+        exe_ids = execution_ids[name]['id'].tolist()
+        for id in exe_ids:
+            artifacts = query.get_all_artifacts_for_execution(id)
+            df = pd.concat([df, artifacts], sort=True, ignore_index=True)
+        df.sort_values("id", inplace=True) 
+        df.drop_duplicates(subset="id",keep='first', inplace=True)
         if df.empty:
             return 
         else:
@@ -50,6 +55,8 @@ def get_all_artifact_ids(mlmdfilepath):
             for art_type in df['type']:
                 filtered_values = df.loc[df['type'] == art_type, ['id', 'name']]
                 artifact_ids[name][art_type] = filtered_values
+    #print("artifact_ids")
+    #print(artifact_ids)
     return artifact_ids
 
 def get_artifacts(mlmdfilepath, pipeline_name, art_type, artifact_ids):