Hi,
batch_size is not a keyword argument in your __init__ function, it is just the second one 
Either call it as (data_source, 3) or in the init function use (self, data_source, batch_size=1, replacement=False).