[QST] 2-stage ensemble throwing 'extra columns' error

❓ Questions & Help

Details

I am trying to simply create a 2-stage ensemble using System and TIS.
I first used NV Tabular to transform my data and trained a DLRM on it and tried to serve workflow + ranking ensemble using TIS, but it was giving me an error saying there are two extra columns (userID and itemID) and won't allow me to do inference, what have I done wrong ?

The transform workflow is defined as in the image. I did not 'categorify' user_id and item_id, because those were tokenized beforehand and I just want to use the original tokens as features and avoid an extra mapping step in the end (as long as I understand right, the categorify is not equal to 'tokenize', instead it is doing some kind of 'binning' operation on the features, right?)

the schema looks like this:

Model info looks like this:

Start with the attached transform workflow, schema, and model information, then trace how the 2-stage ensemble passes columns from NV Tabular through DLRM and TIS. Determine why userID and itemID are treated as extra during inference and document the configuration or input-schema change needed for the workflow to run.