Save ML model in the registry fails

pablobd — Thu, 04 Jan 2024 15:09:15 GMT

Using `mlflow.pyfunc.log_model` to save a model in the model registry fails with the following error: `PicklingError: Can't pickle <built-in function input>: it's not the same object as builtins.input`

I've reproduced it locally and got to the conclusion that it comes because I am using `boto3.resource("dynamodb")` - this is because DynamoDB provides features at inference time. Using the online feature store (also with DynamoDB) may work but it's not my preferred solution.

Do you know any workaround?

Thanks

topic Save ML model in the registry fails in Machine Learning

Save ML model in the registry fails