fix: adding fillter by studyId

opentargets · Apr 2, 2024 · 1dc6701 · 1dc6701
1 parent 67f7f36
commit 1dc6701
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"))