1  from pyspark.ml.param import Param as SparkMLParam
 2  from pyspark.ml.util import Identifiable
 3  
 4  from mlflow.entities import Param
 5  
 6  
 7  def test_spark_integration():
 8      key = SparkMLParam(Identifiable(), "name", "doc")
 9      value = 123
10      param = Param(key, value)
11      assert param.key == "name"
12      assert param.value == "123"