def __init__(self, idx):
    """Initialize the Edge.

    Args:
      idx: The index of the Edge.
    """
    self.idx = idx
    self.source = -1
    self.target = -1
    self.data = {}
Initialize the Edge.

    Args:
      idx: The index of the Edge.
    
tensor2tensor/insights/graph.py