Optionally disable logging in the data sampler to support predict_step
stopping this for now until our CI is out of maintenance  Looks good, thanks! Can we clean up some of the old comments in the test script?
Pytorch lightning's `predict_step` does not support logging on the module. This PR adds an option to the data sampler to disable logging which allows the predict step to work.