Only single lgx reductions

- For `lgx` reduction, tracking of duplicate alleles is removed. For duplicate alleles, map it to the lower numbered allele in the list.

Author: pbashyal-nmdp

pyard/ard.py

@@ -246,9 +246,7 @@ def _redux_allele(
         elif redux_type == "P" and allele in self.ars_mappings.p_group:
             return self.ars_mappings.p_group[allele]
         elif redux_type in ["lgx", "lg"]:
-            if allele in self.ars_mappings.dup_lgx:
-                redux_allele = self.ars_mappings.dup_lgx[allele]
-            elif allele in self.ars_mappings.lgx_group:
+            if allele in self.ars_mappings.lgx_group:
                 redux_allele = self.ars_mappings.lgx_group[allele]
             else:
                 # for 'lgx' or 'lg' mode when allele is not in G group,

pyard/data_repository.py

@@ -50,6 +50,7 @@
     get_1field_allele,
 )
 from .serology import broad_splits_dna_mapping, SerologyMapping
+from .smart_sort import smart_sort_comparator
 
 
 def expression_reduce(df):
@@ -113,15 +114,6 @@ def generate_ard_mapping(db_connection: sqlite3.Connection, imgt_version) -> ARS
     mlgx = df_g_group.drop_duplicates(["2d", "lgx"])["2d"].value_counts()
     multiple_lgx_list = mlgx[mlgx > 1].index.to_list()
 
-    # Keep only the alleles that have more than 1 mapping
-    dup_lgx = (
-        df_g_group[df_g_group["2d"].isin(multiple_lgx_list)][["lgx", "2d"]]
-        .drop_duplicates()
-        .groupby("2d", as_index=True)
-        .agg("/".join)
-        .to_dict()["lgx"]
-    )
-
     # Extract G group mapping
     df_g = pd.concat(
         [
@@ -154,6 +146,25 @@ def generate_ard_mapping(db_connection: sqlite3.Connection, imgt_version) -> ARS
     )
     lgx_group = df_lgx.set_index("A")["lgx"].to_dict()
 
+    # Find the alleles that have more than 1 mapping
+    dup_lgx = (
+        df_g_group[df_g_group["2d"].isin(multiple_lgx_list)][["lgx", "2d"]]
+        .drop_duplicates()
+        .groupby("2d", as_index=True)
+        .agg(list)
+        .to_dict()["lgx"]
+    )
+    print(dup_lgx)
+    # Do not keep duplicate alleles for lgx. Issue #333
+    # DPA1*02:02/DPA1*02:07 ==> DPA1*02:02
+    #
+    lowest_numbered_dup_lgx = {
+        k: sorted(v, key=functools.cmp_to_key(smart_sort_comparator))[0]
+        for k, v in dup_lgx.items()
+    }
+    # Update the lgx_group with the allele with the lowest number
+    lgx_group.update(lowest_numbered_dup_lgx)
+
     # Extract exon mapping
     df_exon = pd.concat(
         [
@@ -164,7 +175,6 @@ def generate_ard_mapping(db_connection: sqlite3.Connection, imgt_version) -> ARS
 
     ars_mapping = ARSMapping(
         dup_g=dup_g,
-        dup_lgx=dup_lgx,
         g_group=g_group,
         p_group=p_group,
         lgx_group=lgx_group,

pyard/db.py

@@ -461,9 +461,6 @@ def set_user_version(connection: sqlite3.Connection, version: int):
 
 def load_ars_mappings(db_connection):
     dup_g = load_dict(db_connection, table_name="dup_g", columns=("allele", "g_group"))
-    dup_lgx = load_dict(
-        db_connection, table_name="dup_lgx", columns=("allele", "lgx_group")
-    )
     g_group = load_dict(db_connection, table_name="g_group", columns=("allele", "g"))
     p_group = load_dict(db_connection, table_name="p_group", columns=("allele", "p"))
     lgx_group = load_dict(
@@ -475,7 +472,6 @@ def load_ars_mappings(db_connection):
     p_not_g = load_dict(db_connection, table_name="p_not_g", columns=("allele", "lgx"))
     return ARSMapping(
         dup_g=dup_g,
-        dup_lgx=dup_lgx,
         g_group=g_group,
         p_group=p_group,
         lgx_group=lgx_group,
@@ -497,12 +493,6 @@ def save_ars_mappings(db_connection: sqlite3.Connection, ars_mapping: ARSMapping
         dictionary=ars_mapping.dup_g,
         columns=("allele", "g_group"),
     )
-    save_dict(
-        db_connection,
-        table_name="dup_lgx",
-        dictionary=ars_mapping.dup_lgx,
-        columns=("allele", "lgx_group"),
-    )
     save_dict(
         db_connection,
         table_name="g_group",

pyard/mappings.py

@@ -23,7 +23,6 @@
 
 ars_mapping_tables = [
     "dup_g",
-    "dup_lgx",
     "g_group",
     "p_group",
     "lgx_group",

tests/features/p_g_group.feature

@@ -90,10 +90,10 @@ Feature: P and G Groups
       | C*02:10    | lg    | C*02:02g     |
       | C*02:10    | lgx   | C*02:02      |
 
-    Examples: lgx with duplicates
-      | Allele        | Level | Redux Allele            |
-      | DPA1*02:12    | lgx   | DPA1*02:02/DPA1*02:07   |
-      | DPA1*02:12    | lg    | DPA1*02:02g/DPA1*02:07g |
-      | DQA1*03:03    | lgx   | DQA1*03:01              |
-      | DQA1*03:03    | lg    | DQA1*03:01g             |
-      | DQA1*03:03:09 | lg    | DQA1*03:03g             |
+    Examples: lgx redux with duplicate G groups
+      | Allele        | Level | Redux Allele |
+      | DPA1*02:12    | lgx   | DPA1*02:02   |
+      | DPA1*02:12    | lg    | DPA1*02:02g  |
+      | DQA1*03:03    | lgx   | DQA1*03:01   |
+      | DQA1*03:03    | lg    | DQA1*03:01g  |
+      | DQA1*03:03:09 | lg    | DQA1*03:03g  |