test: fixing doctest

opentargets · Apr 4, 2024 · ba50ed6 · ba50ed6
1 parent e91d03f
commit ba50ed6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/gentropy/datasource/gwas_catalog/associations.py b/src/gentropy/datasource/gwas_catalog/associations.py
@@ -52,7 +52,7 @@ def convert_gnomad_position_to_ensembl(
         Examples:
             >>> d = [(1, "A", "C"), (2, "AA", "C"), (3, "A", "AA")]
             >>> df = spark.createDataFrame(d).toDF("position", "reference", "alternate")
-            >>> df.withColumn("new_position", convert_gnomad_position_to_ensembl(f.col("position"), f.col("reference"), f.col("alternate"))).show()
+            >>> df.withColumn("new_position", GWASCatalogCuratedAssociationsParser.convert_gnomad_position_to_ensembl(f.col("position"), f.col("reference"), f.col("alternate"))).show()
             +--------+---------+---------+------------+
             |position|reference|alternate|new_position|
             +--------+---------+---------+------------+
@@ -61,7 +61,6 @@ def convert_gnomad_position_to_ensembl(
             |       3|        A|       AA|           4|
             +--------+---------+---------+------------+
             <BLANKLINE>
-
         """
         return f.when(
             (f.length(reference) > 1) | (f.length(alternate) > 1), position + 1