fix: avoid empty step_idx list when n_steps==seq_len;

fix: avoid empty step_idx list when n_steps==seq_len; #89