From 1dc67013585bcaba5a291ada1cdc05422a59c79a Mon Sep 17 00:00:00 2001 From: Yakov Tsepilov Date: Tue, 2 Apr 2024 17:01:49 +0100 Subject: [PATCH] fix: adding fillter by studyId --- src/gentropy/susie_finemapper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/gentropy/susie_finemapper.py b/src/gentropy/susie_finemapper.py index e6d577784..001f34bbf 100644 --- a/src/gentropy/susie_finemapper.py +++ b/src/gentropy/susie_finemapper.py @@ -65,7 +65,8 @@ def susie_finemapper_chr_pos( GWAS_df = GWAS._df GWAS_df = GWAS_df.filter( - (f.col("chromosome") == chromosome) + (f.col("studyId") == _studyId) + & (f.col("chromosome") == chromosome) & (f.col("position") >= position - (window / 2)) & (f.col("position") <= position + (window / 2)) ).withColumn("z", f.col("beta") / f.col("standardError"))