Hi all! Using

kedro-mlflow

is it possible to retrieve a model from S3 based on it’s name and stage, directly from a catalog entry? E.g., I’d like to get the

production

version of the

foobar

model. The

.pkl

is saved as an artifact on S3. I was thinking of building a custom dataset that calls the Mlflow REST API, get’s the path to the model with the given name and stage, and then downloads the artifact from S3, but I wonder if there is a simpler/better way of doing it.

I guess a custom dataset is a way to go here. Summon @Yolan Honoré-Rougé to confirm 🧙

The

MlflowModelRegistryDataSet

is the way to go ;)

Awesome! Will take a look. Thanks a lot, folks