From 8c896100ef302d1a1e441163aca138d95366edb6 Mon Sep 17 00:00:00 2001
From: Lucas Nestler <39779310+ClashLuke@users.noreply.github.com>
Date: Tue, 12 May 2026 17:48:27 +0200
Subject: [PATCH] Fix log access for target_key in sweep.py

---
 pufferlib/sweep.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pufferlib/sweep.py b/pufferlib/sweep.py
index 1212d8e1ec..9e237778c2 100644
--- a/pufferlib/sweep.py
+++ b/pufferlib/sweep.py
@@ -951,7 +951,7 @@ def early_stop(self, logs, target_key):
                 logs['is_loss_nan'] = True
                 return True
 
-        if 'uptime' not in logs or target_key not in logs:
+        if 'uptime' not in logs or target_key not in logs['env']:
             return False
 
         metric_val, cost = logs['env'][target_key], logs['uptime']