Skip to content

Conversation

@oliverholworthy
Copy link
Contributor

@oliverholworthy oliverholworthy commented Jun 6, 2023

Follow-up to #354

  • Adds assersion for expected dtype returned by an NVTabular workflow running in Triton
  • Coerces offsets to int32 in match_representations

Fixes error we're seeing in an ensemble that combines NVTabular workflow with ragged outputs and a Pytorch model. In this case the offsets are returning as int64, while the model expects int32 for these.

inference input data-type is 'INT64', model expects 'INT32' for '1_predictpytorchtriton'

@oliverholworthy oliverholworthy added the chore Maintenance for the repository label Jun 6, 2023
@oliverholworthy oliverholworthy added this to the Merlin 23.06 milestone Jun 6, 2023
@oliverholworthy oliverholworthy self-assigned this Jun 6, 2023
@oliverholworthy oliverholworthy marked this pull request as ready for review June 6, 2023 17:16
@github-actions
Copy link

github-actions bot commented Jun 6, 2023

Documentation preview

https://nvidia-merlin.github.io/systems/review/pr-361

@nv-alaiacano nv-alaiacano merged commit ed7009c into NVIDIA-Merlin:main Jun 6, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

chore Maintenance for the repository

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants