spidey121 commited on
Commit
d7dfb72
·
1 Parent(s): 36b96c0

fix round 1

Browse files
Files changed (1) hide show
  1. inference.py +6 -3
inference.py CHANGED
@@ -80,9 +80,12 @@ def run_task(task_name):
80
  pass
81
 
82
  # Model chooses action
83
- action = choose_action(client, state)
84
-
85
- # fallback
 
 
 
86
  if action not in env.action_space():
87
  action = "detect_attack"
88
 
 
80
  pass
81
 
82
  # Model chooses action
83
+
84
+ action = choose_action(client, state)
85
+ if random.random() < 0.2:
86
+ action = random.choice(env.action_space())
87
+
88
+ # fallback
89
  if action not in env.action_space():
90
  action = "detect_attack"
91