Revert "gnn"

This reverts commit 124901d173f38b733cfda7fae72d5fa5653693d1.
Revert "Revert "gnn""

This reverts commit ccd2e2f2b0cbeb7551ebcf635397bf3596fa690c.