wip - batch input to partial fit