Fix: update tasks/rate_limit/grader.py

tasks/rate_limit/grader.py

@@ -1,23 +1,18 @@
 """Grader for rate_limit task: 429 Too Many Requests."""
 
-import sys
-from pathlib import Path
-
-_project_root = str(Path(__file__).parent.parent.parent)
-if _project_root not in sys.path:
-    sys.path.insert(0, _project_root)
-
-from environment.api_triage_env import APITriageEnv
-from environment.incident_generator import get_incident_by_type
-
 
 def grade() -> float:
-    """Grade the rate_limit task by simulating an optimal agent.
-
-    Returns:
-        Score strictly between 0 and 1.
-    """
+    """Grade the rate_limit task. Returns float strictly between 0 and 1."""
     try:
+        import sys
+        from pathlib import Path
+        _root = str(Path(__file__).resolve().parent.parent.parent)
+        if _root not in sys.path:
+            sys.path.insert(0, _root)
+
+        from environment.api_triage_env import APITriageEnv
+        from environment.incident_generator import get_incident_by_type
+
         env = APITriageEnv(max_steps=10)
         env.incident = get_incident_by_type("rate_limit")
         if env.incident is None:
@@ -27,15 +22,10 @@ def grade() -> float:
         env.step_counter = 0
         env.total_reward = 0.0
 
-        # Optimal sequence: inspect → correct fix → resolve
-        actions = ["inspect_logs", "wait_retry", "resolve"]
-        for action in actions:
-            state, reward, done, info = env.step(action)
+        for action in ["inspect_logs", "wait_retry", "resolve"]:
+            _, _, done, info = env.step(action)
             if done:
-                if info.get("resolution") == "success":
-                    return 0.95
-                else:
-                    return 0.1
+                return 0.95 if info.get("resolution") == "success" else 0.1
         return 0.1
     except Exception:
         return 0.1