def METHOD_NAME(ds):
    feed_dict = {}
    feed_dict["batch_nodes"] = np.array(ds.nodes_idx, dtype="int64")
    feed_dict["labels"] = np.array(ds.labels, dtype="int64")
    def r():
        yield feed_dict
    return r