fix: do not infer features_list from df

opentargets · Feb 13, 2025 · c635e18 · c635e18
1 parent c388ff5
commit c635e18
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gentropy/dataset/l2g_prediction.py b/src/gentropy/dataset/l2g_prediction.py
@@ -95,7 +95,8 @@ def from_credible_set(
                     .select("studyLocusId")
                     .join(feature_matrix._df, "studyLocusId")
                     .filter(f.col("isProteinCoding") == 1)
-                )
+                ),
+                features_list=l2g_model.features_list,
             )
             .fill_na()
             .select_features(l2g_model.features_list)