format

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