KeyError: "['XXX'] not in index"

[njkrichardson]

I added the @profile decorator to a custom module method (after from pytorch_memlab import LineProfiler, profile) and the output is the truncated display with the error KeyError: "['XXX'] not in index" for my method XXX. What's the meaning of this?

[Stonesjtu]

Can you post the code snippet and the full error message? The error message indicates the function under profiling is missing. However I still need more information to find out the actual reason.