Spaces:

junaid0600
/

sql-db-engineer-agent

Sleeping

junaid0600 commited on Apr 9

Commit

0ac8fe8

1 Parent(s): 81abf49

fixed new error

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -208,7 +208,7 @@ def run_episode(client: OpenAI, difficulty: str, task_id: str) -> dict:
         # Calculate score
         total_reward = sum(rewards)
-        score        = min(max(total_reward / MAX_STEPS, 0.0), 1.0)
         success      = score >= SUCCESS_SCORE_THRESHOLD
     except Exception as e:

         # Calculate score
         total_reward = sum(rewards)
+        score        = min(max(total_reward / MAX_STEPS, 0.001), 0.999)
         success      = score >= SUCCESS_SCORE_THRESHOLD
     except Exception as e: