snp_to_observations

Gregor-Mendel-Institute · May 9, 2021 · f16e98a · f16e98a
1 parent b33ada0
commit f16e98a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/snpmatch/core/infer.py b/snpmatch/core/infer.py
@@ -198,7 +198,7 @@ def snp_to_observations(input_snps, snpsP1_gt, snpsP2_gt, polarize = None):
         else:
             ebPolarised[np.where((np.equal( input_snps_mask, snpsP1_gt_mask )) )[0] ] = 0  ## 00
             ebPolarised[np.where((np.equal( input_snps_mask, snpsP2_gt_mask )) )[0] ] = 2  ## 11
-        ebPolarised[np.where( np.equal( input_snps_mask, np.repeat(2, num_snps)) | (snpsP1_gt_mask != snpsP2_gt_mask ) )[0] ] = 1
+        ebPolarised[np.where( np.equal( input_snps_mask, np.repeat(2, num_snps)) & (snpsP1_gt_mask != snpsP2_gt_mask ) )[0] ] = 1
         return(ebPolarised)