Flip semsimsearch target enum for better readability

monarch-initiative · Sep 21, 2023 · d67f60c · d67f60c
1 parent a70bfe3
commit d67f60c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/backend/src/monarch_py/implementations/oak/oak_implementation.py b/backend/src/monarch_py/implementations/oak/oak_implementation.py
@@ -14,12 +14,12 @@
 
 
 class SemsimSearchCategory(Enum):
-    HUMAN_GENE = "HGNC"
-    MOUSE_GENE = "MGI"
-    RAT_GENE = "RGD"
-    ZEBRAFISH_GENE = "ZFIN"
-    WORM_GENE = "WB"
-    DISEASE = "MONDO"
+    HGNC = "Human Genes"
+    MGI = "Mouse Genes"
+    RGD = "Rat Genes"
+    ZFIN = "Zebrafish Genes"
+    WB = "C. Elegans Genes"
+    MONDO = "Human Diseases"
 
 
 @dataclass
@@ -87,6 +87,6 @@ def search(self,
             objects=set(objects),
             object_closure_predicates=predicates,
             include_similarity_object=True,
-            subject_prefixes=[target_group.value for target_group in target_groups],
+            subject_prefixes=[target_group.name for target_group in target_groups],
             limit=limit,
         )