From 2af1074def5d6e02838c2188179270ec72b2cee2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Irene=20L=C3=B3pez=20Santiago?=
 <45119610+ireneisdoomed@users.noreply.github.com>
Date: Tue, 5 Nov 2024 14:18:02 +0000
Subject: [PATCH] feat(feature_matrix): extract features for gwas associations
 only (#901)

---
 src/gentropy/l2g.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/gentropy/l2g.py b/src/gentropy/l2g.py
index cc1b5a5d1..b585166c1 100644
--- a/src/gentropy/l2g.py
+++ b/src/gentropy/l2g.py
@@ -84,7 +84,9 @@ def __init__(
             gene_index=gene_index,
         )
 
-        fm = credible_set.build_feature_matrix(features_list, features_input_loader)
+        fm = credible_set.filter(f.col("studyType") == "gwas").build_feature_matrix(
+            features_list, features_input_loader
+        )
         fm._df.write.mode(session.write_mode).parquet(feature_matrix_path)