fix l2g_feature_matrix tests

opentargets · Dec 11, 2024 · 6ac2d12 · 6ac2d12
1 parent 921c820
commit 6ac2d12
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/gentropy/dataset/test_l2g_feature_matrix.py b/tests/gentropy/dataset/test_l2g_feature_matrix.py
@@ -26,6 +26,9 @@
 from gentropy.dataset.study_locus import StudyLocus
 from gentropy.method.l2g.feature_factory import L2GFeatureInputLoader
 
+if TYPE_CHECKING:
+    from pyspark.sql import SparkSession
+
 
 def test_select_features_inheritance(
     spark: SparkSession, mock_l2g_feature_matrix: L2GFeatureMatrix