immortalindeed commited on
Commit
ddafe29
·
1 Parent(s): 723407b

Remove mock.py (old debug file with wrong [END] format)

Browse files
Files changed (1) hide show
  1. mock.py +0 -27
mock.py DELETED
@@ -1,27 +0,0 @@
1
- import sys
2
- sys.path.append('.')
3
- from inference import parse_action, MAX_STEPS
4
- import functools
5
-
6
- # Mock the print function so we can capture exactly what the user wants to see
7
- def run_mock():
8
- task_id = "sec_easy"
9
- episode_id = "ep-abc12345"
10
- step_num = 1
11
- display_action = "identify_vulnerability"
12
- reward = 0.5
13
- done = False
14
-
15
- print(f'[START] task_id={task_id} episode_id={episode_id}')
16
- print(f'[STEP] task_id={task_id} step={step_num} action={display_action} reward={reward:.4f} done={done}')
17
-
18
- step_num = 2
19
- display_action = "propose_fix"
20
- reward = 1.0
21
- done = True
22
- print(f'[STEP] task_id={task_id} step={step_num} action={display_action} reward={reward:.4f} done={done}')
23
-
24
- total_reward = 1.5
25
- print(f'[END] task_id={task_id} episode_id={episode_id} total_reward={total_reward:.4f} steps={step_num}')
26
-
27
- run_mock()