From 3158933e200e849adbb10aff62a76f8718615840 Mon Sep 17 00:00:00 2001 From: sergiopaniego Date: Fri, 30 Oct 2026 18:27:35 +0100 Subject: [PATCH] Require --env-url in the Wordle GRPO script --- examples/wordle/evaluate.py | 3 ++- examples/wordle/grpo.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/wordle/evaluate.py b/examples/wordle/evaluate.py index 147ee15ac..b52199dfa 100644 --- a/examples/wordle/evaluate.py +++ b/examples/wordle/evaluate.py @@ -200,7 +200,8 @@ def main(): ) model = AutoModelForCausalLM.from_pretrained(args.model, dtype=dtype).to(device) # A private Space needs your token. A public one ignores it. - headers = {"Authorization": f"Bearer {get_token()}"} if get_token() else None + token = get_token() + headers = {"Authorization": f"Bearer {token}"} if token else None if args.secrets: secrets = args.secrets.split(",") diff --git a/examples/wordle/grpo.py b/examples/wordle/grpo.py index 3b48fff61..c31fa9d89 100644 --- a/examples/wordle/grpo.py +++ b/examples/wordle/grpo.py @@ -65,13 +65,14 @@ def main(): parser = argparse.ArgumentParser() parser.add_argument("--model", required=True) parser.add_argument("--hub-model-id", required=True) - parser.add_argument("--env-url", default="https://openenv-wordle.hf.space") + parser.add_argument("--env-url", required=True) parser.add_argument("--max-steps", type=int, default=150) parser.add_argument("--trackio-space-id", default=None) args = parser.parse_args() env_url = args.env_url # A private Space needs your token. A public one ignores it. - headers = {"Authorization": f"Bearer {get_token()}"} if get_token() else None + token = get_token() + headers = {"Authorization": f"Bearer {token}"} if token else None class WordleEnv: def __init__(self):