def METHOD_NAME(epochs):
    for __ in range(epochs):
        for batch in dataset.METHOD_NAME(batch_size=s,
                                         pad_batches=True):
            adjacency_tensor = one_hot(batch[0], gan.edges)
            node_tesor = one_hot(batch[1], gan.nodes)
            yield {
                gan.data_inputs[0]: adjacency_tensor,
                gan.data_inputs[1]: node_tesor
            }