Clean up sample loop, pass discard_sample keyword as kwarg to step/step_warmup - #194
Merged
sample loop, pass discard_sample keyword as kwarg to step/step_warmup - #194