Cannot serialize this model error when attempting MLFlow for SparkNLP

[Slalom_Tobias]

I'm attempting to use MLFlow to register models in Databricks and am following the recipe at...

https://nlp.johnsnowlabs.com/docs/en/licensed_serving_spark_nlp_via_api_databricks_mlflow

when i execute...

mlflow.spark.log_model(pipeline, "lemmatizer", conda_env=conda_env)

i get the following error....

MlflowException: Cannot serialize this model. MLflow can only save descendants of pyspark.Modelthat implement MLWritable and MLReadable.

can anyone help?

[Vidula]

Hi @Tobias Cortese​ 

Hope all is well! Just wanted to check in if you were able to resolve your issue and would you be happy to share the solution or mark an answer as best? Else please let us know if you need more help. 

We'd love to hear from you.

Thanks!

[Slalom_Tobias]

I haven't found a solution within the documentation.