Could you post an executable code snippet to reproduce this issue, please?
You don’t need to provide the real data. Just random values with the right type and shape should be sufficient to run the code.