cosmit: I think the second part of the expression can fit on one 80 char line:

```python
        batch_slices = list(gen_even_slices(
            n_batches * self.batch_size, n_batches))
```