在pyspark/jupyter env中用MOJO得分

我想用H2O的mojo对象在Pyspark/Jupyter笔记本中为传入的数据管道打分

from pyspark.sql import SparkSession
spark = SparkSession.builder \
                    .config("spark.executor.extraClassPath","/path/h2o-genmodel.jar") \
                    .config("spark.driver.extraClassPath","/path/h2o-genmodel.jar") \
                    .config("spark.jars","/path/h2o-genmodel.jar")
MojoModel = spark._jvm.hex.genmodel.MojoModel
EasyPredictModelWrapper = spark._jvm.hex.genmodel.easy.EasyPredictModelWrapper
RowData = spark._jvm.hex.genmodel.easy.Rowdata
mojo = MojoModel.load("mojo_path")

当我跑步时：

print(spark._jvm.hex.genmodel.MojoModel)

我得到“0x7fb8ac768d30处的py4j.java_gateway.JavaPackage对象”。这是否告诉我spark已成功加载并查看该包

然而，该命令

mojo = MojoModel.load("mojo_path")

给出“TypeError:'JavaPackage'对象不可调用”

请就其他依赖项或配置设置提供建议。我选择的配置设置基于其他几个stackoverflow票证