format

Signed-off-by: Weichen Xu <[email protected]>
WeichenXu123 · Jul 22, 2024 · 22e5c7b · 22e5c7b
1 parent 6e3be05
commit 22e5c7b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/pyspark/ml/util.py b/python/pyspark/ml/util.py
@@ -465,9 +465,9 @@ def saveMetadata(
             instance, sc, extraMetadata, paramMap
         )
         spark = SparkSession.getActiveSession()
-        spark.createDataFrame([(metadataJson,)], schema=["value"]).coalesce(1).write.text(
-            metadataPath
-        )
+        spark.createDataFrame(  # type: ignore[union-attr]
+            [(metadataJson,)], schema=["value"]
+        ).coalesce(1).write.text(metadataPath)
 
     @staticmethod
     def _get_metadata_to_save(
@@ -581,7 +581,7 @@ def loadMetadata(path: str, sc: "SparkContext", expectedClassName: str = "") ->
         """
         metadataPath = os.path.join(path, "metadata")
         spark = SparkSession.getActiveSession()
-        metadataStr = spark.read.text(metadataPath).first()[0]
+        metadataStr = spark.read.text(metadataPath).first()[0]  # type: ignore[union-attr,index]
         loadedVals = DefaultParamsReader._parseMetaData(metadataStr, expectedClassName)
         return loadedVals