Source code for proton_decay_study.generators.base

import logging
from abc import abstractmethod, ABCMeta


class BaseDataGenerator(object):
  """
    Base data generator which hooks into the networks to provide
    an interface to the incoming data.
  """
  logger = logging.getLogger('pdk.generator')
  __metaclass__ = ABCMeta

  def __init__(self):
    self._dataset = None

  def __len__(self):
    return len(self._dataset)

  def __iter__(self):
    return self

  def __next__(self):
    return self.next()

  @abstractmethod
  def next(self):
    raise StopIteration()