mlfow.spark.log_model fails with input_sample

I’m trying to log my ALS model with mlflow.spark.log_model. my data looks like this (movielens dataset)