MIA-GCL / MVGRL / GCL2 / augmentors / identity.py
identity.py
Raw
from GCL2.augmentors.augmentor import Graph, Augmentor


class Identity(Augmentor):
    def __init__(self):
        super(Identity, self).__init__()

    def augment(self, g: Graph) -> Graph:
        return g