fixed dataset and cohorts response by id

GenomicDataInfrastructure · Jul 19, 2023 · 213867b · 213867b
1 parent 0e7e8e6
commit 213867b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/beacon/response/build_response.py b/beacon/response/build_response.py
@@ -139,6 +139,14 @@ def build_beacon_resultset_response_by_dataset(data,
                         elif doc['individualId'] in dataset_dict['ids'][0]['individualIds']:
                             dataset_id = dataset_dict['dataset']
                             response_dict[dataset_id].append(doc)
+                elif str(entity_schema) == 'DefaultSchemas.DATASETS':
+                        if doc['id'] == dataset_dict['dataset']:
+                            dataset_id = dataset_dict['dataset']
+                            response_dict[dataset_id].append(doc)
+                elif str(entity_schema) == 'DefaultSchemas.COHORTS':
+                        if doc['id'] == dataset_dict['dataset']:
+                            dataset_id = dataset_dict['dataset']
+                            response_dict[dataset_id].append(doc)
                 else:
                         if doc['id'] in dataset_dict['ids'][0]['biosampleIds']:
                             dataset_id = dataset_dict['dataset']