Cómo especificar la versión de H2O en Sparkling Water?

En un cuaderno de Databricks, estoy tratando de cargar un modelo H2O que fue entrenado para la versión 3.30.1.3 de H2O.

He instalado la versión de Sparkling Water que corresponde a la versión de Spark utilizada para el entrenamiento del modelo (3.0), h2o-pysparkling-3.0, que descargué de PyPI.

El servidor de Sparkling Water está utilizando la última versión de H2O en lugar de la versión que necesito. Tal vez haya una forma de especificar la versión de H2O cuando inicio el contexto de Sparkling Water. Algo como esto:

import h2o
from pysparkling import H2OContext
from pysparkling.ml import H2OBinaryModel

hc = H2OContext.getOrCreate(h2o_version='3.30.1.3')
model = H2OBinaryModel.read('s3://bucket/model_file')

Ejecuto el código anterior sin argumentos en H2OContext.getOrCreate() y obtengo este error:

IllegalArgumentException: 
 El modelo binario ha sido entrenado en H2O versión 3.30.1.3 pero estás ejecutando la versión 3.34.0.6 de H2O.
 Asegúrate de que Sparkling Water/H2O-3 y el modelo binario cargado correspondan a la misma versión de H2O-3.

¿Dónde está la API de Python para Sparkling Water? Si pudiera encontrar eso, podría determinar si hay un argumento de versión de H2O para el inicializador de contexto, pero sorprendentemente hasta ahora me ha sido imposible encontrarlo con Google y explorando en la documentación.

¿O es esto algo que se maneja instalando una versión específica de H2O de Sparkling Water? ¿O tal vez hay otra configuración relevante en algún lugar?