batch_size needs to be here as a kwarg now

else there is no attrib e.args.batch_size