Pipeline inference with Dataset api

Hi, I came across this warning as well. By any chance did you find a solution?