iterator_mixin.py 250 B

12345678910
  1. import numpy as np
  2. import logging
  3. from .base import BaseChainerMixin
  4. from nabirds.utils import new_iterator
  5. class IteratorMixin(BaseChainerMixin):
  6. def new_iterator(self, **kwargs):
  7. self.chainer_check()
  8. return new_iterator(data=self, **kwargs)