format

Signed-off-by: Weichen Xu <[email protected]>

Author: WeichenXu123

python/pyspark/ml/tuning.py

@@ -453,7 +453,7 @@ def load(
                 est = uidToParams[jsonParam["parent"]]
                 param = getattr(est, jsonParam["name"])
 
-                def extract_value(key):
+                def extract_value(key: str) -> Any:
                     if is_saved_by_python_writer:
                         return jsonParam[key]
                     # If the the params are serialized by java writer,