test_assisted_decoding_matches_greedy_search
* fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>