fix import

python/pyspark/sql/pandas/map_ops.py

@@ -23,6 +23,7 @@
 from pyspark.sql.types import StructType
 
 if TYPE_CHECKING:
+    from py4j.java_gateway import JavaObject
     from pyspark.sql.dataframe import DataFrame
     from pyspark.sql.pandas._typing import PandasMapIterFunction, ArrowMapIterFunction
 
@@ -250,7 +251,7 @@ def mapInArrow(
 
     def _build_java_profile(
         self, profile: Optional[ResourceProfile] = None
-    ) -> Optional[JavaObject]:
+    ) -> Optional["JavaObject"]:
         """Build the java ResourceProfile based on PySpark ResourceProfile"""
         from pyspark.sql import DataFrame