I have saved a model in the model registry using MLFlow. How can I find the shap values for this model once I have generated predictions in batch mode? 

Shap tree explainer does not support the mlflow pyfunc model type. When I use mlflow.shap.log_explanation(model.predict, data), I get an error that Provided model function fails when applied to the provided data set. It seems to indicate the required columns are missing. All the required columns were included and when I run predictions with the model it generates predictions. model.predict(data).