This is not related to the API change, but wouldn't "n_steps" be more appropriate than "n_samples"?