If 2 field use lgx_allele_list

Author: pbashyal-nmdp

pyard/ard.py

@@ -275,11 +275,14 @@ def _redux_allele(
         elif redux_type == "S":
             # find serology equivalent in serology_mapping
             if is_2_field_allele(allele):
+                allele = self._redux_allele(allele, "lgx")
                 serology_mapping = db.find_serology_for_allele(
-                    self.db_connection,
-                    allele,
+                    self.db_connection, allele, "lgx_allele_list"
+                )
+            else:
+                serology_mapping = db.find_serology_for_allele(
+                    self.db_connection, allele
                 )
-            serology_mapping = db.find_serology_for_allele(self.db_connection, allele)
             serology_set = set()
             for serology, allele_list in serology_mapping.items():
                 if allele in allele_list.split("/"):

pyard/db.py

@@ -400,7 +400,7 @@ def find_serology_for_allele(
 
     :param connection: db connection of type sqlite.Connection
     :param allele_name: Allele name to use as a prefix to find similar alleles
-    :param column: Column to look for allele
+    :param column: Column to look for allele, "allele_list" or "lgx_allele_list"
     :return: list of similar alleles
     """
     query = f"SELECT serology, {column} FROM serology_mapping WHERE {column} LIKE ?"