fix: susie credible sets with unknown confidence

opentargets · Oct 18, 2024 · fa824d7 · fa824d7
1 parent 333facb
commit fa824d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gentropy/dataset/study_locus.py b/src/gentropy/dataset/study_locus.py
@@ -1175,7 +1175,7 @@ def assign_confidence(self: StudyLocus) -> StudyLocus:
  df = self.df.withColumn(
  "confidence",
  f.when(
- (f.col("finemappingMethod") == "SuSiE-inf")
+ (f.col("finemappingMethod").isin(["SuSiE-inf", "SuSiE"]))
  & (
  ~f.array_contains(
  f.col("qualityControls"),
@@ -1185,7 +1185,7 @@ def assign_confidence(self: StudyLocus) -> StudyLocus:
  CredibleSetConfidenceClasses.FINEMAPPED_IN_SAMPLE_LD.value,
  )
  .when(
- (f.col("finemappingMethod") == "SuSiE-inf")
+ (f.col("finemappingMethod").isin(["SuSiE-inf", "SuSiE"]))
  & (
  f.array_contains(
  f.col("qualityControls"),