https://kedro.org/ logo
#plugins-integrations
Title
# plugins-integrations
r

Rennan Haro

11/07/2023, 2:02 PM
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.
m

marrrcin

11/07/2023, 2:08 PM
I guess a custom dataset is a way to go here. Summon @Yolan Honoré-Rougé to confirm 🧙
y

Yolan Honoré-Rougé

11/07/2023, 2:34 PM
The
MlflowModelRegistryDataSet
is the way to go ;)
👍 1
👍🏼 2
You can specify the stage in the load_args
👍🏼 1
r

Rennan Haro

11/07/2023, 3:00 PM
Awesome! Will take a look. Thanks a lot, folks
2 Views