From 7a22a95d29f83363f6b98b0a6fc919a96ad849f6 Mon Sep 17 00:00:00 2001
From: ieuans <ieuanscanlon@hotmail.co.uk>
Date: Wed, 11 Oct 2023 17:48:26 +0100
Subject: [PATCH] Add group to yaml

---
 pyconceptlibraryclient/phenotypes.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pyconceptlibraryclient/phenotypes.py b/pyconceptlibraryclient/phenotypes.py
index 6a4d68a..7a0e385 100644
--- a/pyconceptlibraryclient/phenotypes.py
+++ b/pyconceptlibraryclient/phenotypes.py
@@ -339,6 +339,10 @@ def __format_for_download(self, data: dict | None) -> list | None:
           result[key].append(new_concept)
         continue
 
+      if key == 'group':
+        result[key] = value['id']
+        continue
+
       if isinstance(value, list):
         fields = {k: [v[k] for v in value] for k in value[0]}
         field_keys = fields.keys()