Skip to content

Commit

Permalink
feat(feature_matrix): extract features for gwas associations only (op…
Browse files Browse the repository at this point in the history
  • Loading branch information
ireneisdoomed authored Nov 5, 2024
1 parent 4d8e7c4 commit 2af1074
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/gentropy/l2g.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)


Expand Down

0 comments on commit 2af1074

Please sign in to comment.