Use gene query logic for all gene table values

microbiomedata · Dec 19, 2024 · 204bcd6 · 204bcd6
1 parent 46e369f
commit 204bcd6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/nmdc_server/query.py b/nmdc_server/query.py
@@ -384,7 +384,13 @@ def query(self, db) -> Query:
 
             # Gene function queries are treated differently because they join
             # in three different places (metaT, metaG and metaP).
-            if table == Table.gene_function:
+            if table in [
+                Table.gene_function,
+                Table.kegg_function,
+                Table.go_function,
+                Table.pfam_function,
+                Table.cog_function,
+            ]:
                 metag_matches = filter.matches(db, self.table)
                 metap_conditions = [
                     SimpleConditionSchema(