Merge "Move partition search ML models to a seperate file"