https://github.com/MinkaiXu/GeoDiff/blob/ea0ca48045a2f7abfccd7f0df449e45eb6eae638/models/epsnet/dualenc.py#L210 I am wondering whether this is correct. It seems that `edge_encoder_local` is not called within the whole module.