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

    Args:
      idx: The index of the vertex.
    """
    self.idx = idx
    self.in_edges = []
    self.out_edges = []
Initialize the Vertex.

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