junaid0600 commited on
Commit
e15627e
·
1 Parent(s): 6e703c0

Ensure score never below 0.1 to fix out of range error

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -204,7 +204,7 @@ def run_episode(client: OpenAI, difficulty: str, task_id: str) -> dict:
204
  done = resp.done
205
  obs = resp.observation
206
  except Exception as e:
207
- reward = 0.01
208
  done = False
209
  error_str = str(e)[:100]
210
 
@@ -225,7 +225,7 @@ def run_episode(client: OpenAI, difficulty: str, task_id: str) -> dict:
225
  # Score strictly between 0 and 1 exclusive
226
  if rewards:
227
  total_reward = sum(rewards)
228
- raw_score = total_reward / len(rewards)
229
  else:
230
  raw_score = 0.1
231
 
 
204
  done = resp.done
205
  obs = resp.observation
206
  except Exception as e:
207
+ reward = 0.1
208
  done = False
209
  error_str = str(e)[:100]
210
 
 
225
  # Score strictly between 0 and 1 exclusive
226
  if rewards:
227
  total_reward = sum(rewards)
228
+ raw_score = max(0.1, total_reward / len(rewards))
229
  else:
230
  raw_score = 0.1
231