#389 species fix

antigenomics · Sep 16, 2024 · b07b3d5 · b07b3d5
1 parent a846511
commit b07b3d5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/py_src/runBuidDatabase.py b/py_src/runBuidDatabase.py
@@ -97,6 +97,9 @@
         final_mask = final_mask & mask
         final_mask_alleles = final_mask_alleles & allele_mask
 
+    final_mask_alleles = final_mask_alleles | (master_table['species'] != 'HomoSapiens')
+    final_mask = final_mask | (master_table['species'] != 'HomoSapiens')
+
     master_table.loc[final_mask & final_mask_alleles].set_index('cdr3.alpha').to_csv('../database/vdjdb_full_gene_clear.txt', sep='\t')
     master_table.loc[~final_mask].set_index('cdr3.alpha').to_csv('../database/vdjdb_full_gene_broken.txt', sep='\t')
     master_table.loc[~final_mask_alleles].set_index('cdr3.alpha').to_csv('../database/vdjdb_full_allele_broken.txt', sep='\t')