From 61a05c60471d6a057ed2ad28f64cb20131bd2b70 Mon Sep 17 00:00:00 2001
From: Hanna JULIENNE <hanna.julienne@pasteur.fr>
Date: Tue, 17 Aug 2021 16:13:14 +0200
Subject: [PATCH] Update windows.py
---
raiss/windows.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/raiss/windows.py b/raiss/windows.py
index a1a9b91..bbcb6f4 100644
--- a/raiss/windows.py
+++ b/raiss/windows.py
@@ -164,7 +164,10 @@ def impg_like_imputation(ld_file, ref_panel, zscore, window_size, buffer, lamb,
unknown_in_LD_reg_n_window = in_region(all_unknowns.pos, start_pos, end_pos)
known = zscore.loc[in_LD_reg_n_window].index
+ known = known.intersection(LD_mat.index)
+
unknowns = all_unknowns.loc[unknown_in_LD_reg_n_window].index
+ unknowns = unknowns.intersection(LD_mat.index)
sig_t = LD_mat.loc[known, known]
sig_i_t = LD_mat.loc[unknowns, known]
--
GitLab