add handling for if num disease is not set

monarch-initiative · Jan 5, 2024 · 96e2097 · 96e2097
1 parent c9b2cca
commit 96e2097
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/phenotype2phenopacket/utils/utils.py b/src/phenotype2phenopacket/utils/utils.py
@@ -250,3 +250,5 @@ def filter_diseases(
         return random.choices(
             group_phenotype_annotation(phenotype_annotation_data.df), k=num_disease
         )
+    if num_disease == 0 and omim_id is None and omim_id_list is None:
+        return group_phenotype_annotation(phenotype_annotation_data.df)