Spaces:
Sleeping
Sleeping
Commit ·
e15627e
1
Parent(s): 6e703c0
Ensure score never below 0.1 to fix out of range error
Browse files- 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.
|
| 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
|
| 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 |
|