From f60a424535a7ed4a81cb411e525f896d16e92161 Mon Sep 17 00:00:00 2001 From: Archish Ray <43189420+Archizzle@users.noreply.github.com> Date: Wed, 27 May 2026 01:23:33 -0700 Subject: [PATCH] ensure sweeps use specified metric --- pufferlib/pufferl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pufferlib/pufferl.py b/pufferlib/pufferl.py index 929c300b48..2e0d10beb3 100644 --- a/pufferlib/pufferl.py +++ b/pufferlib/pufferl.py @@ -374,7 +374,7 @@ def _train(env_name, args, sweep_obj=None, result_queue=None, verbose=False): result_queue.put((args['gpu_id'], [match_score], [metrics['uptime'][-1]], [metrics['agent_steps'][-1]])) else: - result_queue.put((args['gpu_id'], metrics['env/score'], metrics['uptime'], metrics['agent_steps'])) + result_queue.put((args['gpu_id'], metrics[target_key], metrics['uptime'], metrics['agent_steps'])) def train(env_name, args=None, gpus=None, **kwargs): args = args or load_config(env_name)