Fixed the pipeline info at the subject level

neurobagel · Oct 3, 2024 · 6cc67f8 · 6cc67f8
1 parent aace3cd
commit 6cc67f8
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/app/api/crud.py b/app/api/crud.py
@@ -217,11 +217,14 @@ async def get(
 
                 subject_data["pipeline"] = subject_data.apply(
                     lambda row: {
-                        name: versions
-                        for name, versions in zip(
-                            row["pipeline_name"], row["pipeline_version"]
+                        name: list(
+                            group.loc[
+                                group["pipeline_name"] == name,
+                                "pipeline_version",
+                            ].unique()
                         )
-                        if pd.notna(name) and pd.notna(versions)
+                        for name in set(row["pipeline_name"])
+                        if pd.notna(name)
                     },
                     axis=1,
                 )