ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'

SOlivero
New Contributor III

I'm using runtime DBR 12.2 LTS ML
Getting this error when running. 

 

import mlflow
logged_model = 'runs:/.../model' #my run id

# Load model as a PyFuncModel.
loaded_model = mlflow.pyfunc.load_model(logged_model)

# Predict on a Pandas DataFrame.
import pandas as pd
loaded_model.predict(pd.DataFrame(data))