cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

Save ML model in the registry fails

pablobd
Contributor II

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

0 REPLIES 0

Join Us as a Local Community Builder!

Passionate about hosting events and connecting people? Help us grow a vibrant local community—sign up today to get started!

Sign Up Now