How to specify query groupings at prediction time in lightgbm when doing learning to rank?

Question

I trained a learning to rank model with lightgbm, in the training datasets the query sizes have to be specified, but in the prediction I couldn\'t