replace NeighborFinder in this tgn by the tgat's NeighborFinder

# Please enter the commit message for your changes. Lines starting
# with '#' will be ignored, and an empty message aborts the commit.
#
# On branch adaptation
# Changes to be committed:
#	modified:   modules/embedding_module.py
#	modified:   train.sh
#
