Anish Shah @ WANDB
09/02/2021, 6:29 PMkedro-mlflow
?
I have trained a custom mock model in which I define operations to save internal data sources to an mlflow run and load them into the desired custom mlflow.pyfunc.PythonModel
.
I am able to serve the model using --no-conda
. Without this flag I get an error raised in which ModuleNotFoundError: No module named '<kedro_package_name>'
. In my MLModel file I set loader_module: <kedro_package_name>.extras.mlflow.loader_cosine_model
. I can also provide any other additional details if anyone can help!