diff --git a/src/gentropy/dataset/colocalisation.py b/src/gentropy/dataset/colocalisation.py index 568b46007..db0040652 100644 --- a/src/gentropy/dataset/colocalisation.py +++ b/src/gentropy/dataset/colocalisation.py @@ -62,7 +62,12 @@ def extract_maximum_coloc_probability_per_region_and_gene( valid_qtls = list( set(EqtlCatalogueStudyIndex.method_to_study_type_mapping.values()) - ) + ) + [ + f"sc{qtl}" + for qtl in set( + EqtlCatalogueStudyIndex.method_to_study_type_mapping.values() + ) + ] if filter_by_qtls: filter_by_qtls = ( diff --git a/src/gentropy/dataset/l2g_features/colocalisation.py b/src/gentropy/dataset/l2g_features/colocalisation.py index c7161dc55..fdbf3ed18 100644 --- a/src/gentropy/dataset/l2g_features/colocalisation.py +++ b/src/gentropy/dataset/l2g_features/colocalisation.py @@ -211,7 +211,7 @@ def compute( """ colocalisation_method = "ECaviar" colocalisation_metric = "clpp" - qtl_type = "eqtl" + qtl_type = ["eqtl", "sceqtl"] return cls( _df=convert_from_wide_to_long( @@ -260,7 +260,7 @@ def compute( """ colocalisation_method = "ECaviar" colocalisation_metric = "clpp" - qtl_type = "eqtl" + qtl_type = ["eqtl", "sceqtl"] return cls( _df=convert_from_wide_to_long( @@ -393,7 +393,7 @@ def compute( """ colocalisation_method = "ECaviar" colocalisation_metric = "clpp" - qtl_types = ["sqtl", "tuqtl"] + qtl_types = ["sqtl", "tuqtl", "scsqtl", "sctuqtl"] return cls( _df=convert_from_wide_to_long( common_colocalisation_feature_logic( @@ -441,7 +441,7 @@ def compute( """ colocalisation_method = "ECaviar" colocalisation_metric = "clpp" - qtl_types = ["sqtl", "tuqtl"] + qtl_types = ["sqtl", "tuqtl", "scsqtl", "sctuqtl"] return cls( _df=convert_from_wide_to_long( common_neighbourhood_colocalisation_feature_logic( @@ -483,7 +483,7 @@ def compute( """ colocalisation_method = "Coloc" colocalisation_metric = "h4" - qtl_type = "eqtl" + qtl_type = ["eqtl", "sceqtl"] return cls( _df=convert_from_wide_to_long( common_colocalisation_feature_logic( @@ -531,7 +531,7 @@ def compute( """ colocalisation_method = "Coloc" colocalisation_metric = "h4" - qtl_type = "eqtl" + qtl_type = ["eqtl", "sceqtl"] return cls( _df=convert_from_wide_to_long( common_neighbourhood_colocalisation_feature_logic( @@ -663,7 +663,7 @@ def compute( """ colocalisation_method = "Coloc" colocalisation_metric = "h4" - qtl_types = ["sqtl", "tuqtl"] + qtl_types = ["sqtl", "tuqtl", "scsqtl", "sctuqtl"] return cls( _df=convert_from_wide_to_long( common_colocalisation_feature_logic( @@ -711,7 +711,7 @@ def compute( """ colocalisation_method = "Coloc" colocalisation_metric = "h4" - qtl_types = ["sqtl", "tuqtl"] + qtl_types = ["sqtl", "tuqtl", "scsqtl", "sctuqtl"] return cls( _df=convert_from_wide_to_long( common_neighbourhood_colocalisation_feature_logic(