fix: tweak spark session builder to fix type hints (#982)

Co-authored-by: Szymon Szyszkowski <[email protected]>
opentargets · Feb 3, 2025 · e2832a3 · e2832a3
1 parent f3a2f9d
commit e2832a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gentropy/common/session.py b/src/gentropy/common/session.py
@@ -42,7 +42,8 @@ def __init__(  # noqa: D107
         )
 
         self.spark = (
-            SparkSession.builder.config(conf=merged_conf)
+            SparkSession.Builder()
+            .config(conf=merged_conf)
             .master(spark_uri)
             .appName(app_name)
             .getOrCreate()