Graph data key error (PinSAGE)

I don’t know why key error occur… could anyone let me know if the data shape is wrong…?

Did you get this error when you run the official PinSAGE example? Did you make any changes to the code?

I’m using my own dataset. but the model code isn’t changed.
Could you guess or advise why this key error occur?
I’m following the sample code which is made with Movielens dataset. I guess the graph data shape ‘g’ is same with sample code.