Fix test_prompt_filtering which got broken after a rebase.#3439

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

igorts/fix_test_rlAI-Hypercomputer/maxtext:igorts/fix_test_rlCopy head branch name to clipboard