pocket representations have different dimensions for pair_repr

Hello,

I tried to generate pocket representations of my own dataset using the code provided in the demo notebook. I got pair representations of different dimensions for different molecules, e.g. [n,n,64], where n is different for different molecules.

On the other hand, when I rerun the demo case, I got the same dimensions for all pair representations.

Could you please help to point out which steps I could have missed?

Thanks!