fix: mhc flag incorrect (opentargets#825)

thehyve · Oct 9, 2024 · 9f0111c · 9f0111c
1 parent b7dce8f
commit 9f0111c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/gentropy/dataset/study_locus.py b/src/gentropy/dataset/study_locus.py
@@ -279,7 +279,8 @@ def validate_lead_pvalue(self: StudyLocus, pvalue_cutoff: float) -> StudyLocus:
             df = self.df.withColumn(
                 qc_colname,
                 create_empty_column_if_not_exists(
-                    qc_colname, get_struct_field_schema(StudyLocus.get_schema(), qc_colname)
+                    qc_colname,
+                    get_struct_field_schema(StudyLocus.get_schema(), qc_colname),
                 ),
             )
         return StudyLocus(
@@ -940,7 +941,7 @@ def qc_MHC_region(self: StudyLocus) -> StudyLocus:
             "qualityControls",
             self.update_quality_flag(
                 f.col("qualityControls"),
-                ~(
+                (
                     (f.col("chromosome") == region.chromosome)
                     & (
                         (f.col("position") <= region.end)