diff --git a/pyard/pyard.py b/pyard/pyard.py index e8099be..a5be8ae 100644 --- a/pyard/pyard.py +++ b/pyard/pyard.py @@ -404,16 +404,26 @@ def redux(self, allele: str, ars_type: str) -> str: return self.dup_g[allele] else: return self.G[allele] - elif ars_type == "lg" and allele in self._lg: - return self.lg[allele] - elif ars_type == "lgx" and allele in self._lgx: - return self.lgx[allele] + elif ars_type == "lg": + if allele in self._lg: + return self.lg[allele] + else: + # for 'lg' when allele is not in G group, + # return allele with only first 2 field + return ':'.join(allele.split(':')[0:2]) + 'g' + elif ars_type == "lgx": + if allele in self._lgx: + return self.lgx[allele] + else: + # for 'lgx' when allele is not in G group, + # return allele with only first 2 field + return ':'.join(allele.split(':')[0:2]) else: if self.remove_invalid: if allele in self.valid: return allele else: - return + return '' else: return allele diff --git a/setup.py b/setup.py index bcc2640..6002b54 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ setup( name='py-ard', - version='0.0.18', + version='0.0.19', description="ARD reduction for HLA with python", long_description=readme + '\n\n' + history, author="CIBMTR",