Show loaded model version in logs

[mck-star-yar]

Description

When I'm using MlflowModelRegistryDataset with the load parameter set to latest logs don't capture the exact version loaded from mlflow. It'd be good to propagate them.

Context

As a DS, looking at the logs I'd like to know what version of model was loaded

Possible Implementation

Ideally, that should live on the mlflow side. But as long as that's not the case we can manually check if user passed the latest arg and log latest version available for the model.

[Galileo-Galilei]

Hi, eventually I could store a bunch of metadata but I also think this should live on mlflow's side. I am a bit reluctant to log too many custom things. I think I can log some metadata of the loaded, but I don't know if I should only log it in hte logs, or in mlflow itself (because one may want to load a model without logging the inference run in mlflow, especially in production). Could yu elaborate more on your ideal result?

[mck-star-yar]

Could yu elaborate more on your ideal result?

Ideal result would be just to show loaded model version, something like: Loaded model version v10