lint

apache · Feb 10, 2025 · 43b4bcb · 43b4bcb
1 parent da5894e
commit 43b4bcb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/pyspark/ml/util.py b/python/pyspark/ml/util.py
@@ -43,6 +43,7 @@
 
 if TYPE_CHECKING:
     from py4j.java_gateway import JavaGateway, JavaObject
+    from pyspark.ml._typing import PipelineStage
     from pyspark.ml.base import Params
     from pyspark.ml.wrapper import JavaWrapper
     from pyspark.core.context import SparkContext