Glad you like it ! If you are interested in custom model to wrap an entire kedro pipeline (i.e. including pre and post processing, and artefacts other then the model like an encoder/a tokenizer), have a look at "kedro mlflow modelify" command and "pipeline_ml_factory" function. I'd love getting feedback on this and find out if people find it useful.