Hi @Kumaran ! Thank you for this response! Unfortunately, I find that this same thing does not work with a Catboost Model, event though mlflow.catboost flavour is supported by MLFlow. Could you help me with this?
These are the libs I'm using:

%pip install 'catboost==1.2.5' -q 
%pip install 'databricks-feature-engineering==0.6.0' -q 
%pip install 'mlflow==2.14.3' -q 
%pip install 'shap==0.44.0' -q

I log the model with:

with mlflow.start_run():
  fe.log_model(
    model = model, 
    artifact_path = 'model',
    flavor = mlflow.catboost,
    training_set = training_set,
    registered_model_name = model_uc_name,
    signature = signature,
    input_example = X_train.head(1)
)

I load it like this:

best_model = mlflow.catboost.load_model(model_uri)

 And I get this error:

MlflowException: Model does not have the "catboost" flavor.

And I need to use the FE client to use your cool Feature Lookups. Please help, I'd really apreciate it!

Cheers!!