fix: reduced the number of scenario in inference

inference.py

@@ -48,9 +48,9 @@ TEMPERATURE      = 0.3
 MAX_TOKENS       = 256
 SUCCESS_THRESHOLD = 0.5
 
-EASY_SCENARIOS   = [k for k, v in SCENARIOS.items() if v["difficulty"] == "easy"][:2]
-MEDIUM_SCENARIOS = [k for k, v in SCENARIOS.items() if v["difficulty"] == "medium"][:2]
-HARD_SCENARIOS   = [k for k, v in SCENARIOS.items() if v["difficulty"] == "hard"][:2]
+EASY_SCENARIOS   = [k for k, v in SCENARIOS.items() if v["difficulty"] == "easy"][:1]
+MEDIUM_SCENARIOS = [k for k, v in SCENARIOS.items() if v["difficulty"] == "medium"][:1]
+HARD_SCENARIOS   = [k for k, v in SCENARIOS.items() if v["difficulty"] == "hard"][:1]
 
 SYSTEM_PROMPT = textwrap.dedent("""
     You are a machine learning engineer diagnosing a failed training run.