import tensorflow as tf
import numpy as np


class Dataset:
    def __init__(self, data, epoch, batch_size) -> None:
        return

    def __call__(self):
        """ return data as an iterator """
        dataset = self
        return dataset
