da03 commited on
Commit
21ceee1
·
1 Parent(s): a1aa2aa
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -221,7 +221,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
221
 
222
 
223
  for action_type, pos in previous_actions: #[-8:]:
224
- print ('here3', action_type, pos)
225
  if action_type == 'move':
226
  action_type = 'N'
227
  if action_type == 'left_click':
@@ -439,9 +439,9 @@ async def websocket_endpoint(websocket: WebSocket):
439
 
440
  action_type = data.get("action_type")
441
  mouse_position = data.get("mouse_position")
442
- if np.random.random() < 0.5:
443
  print ('setting left click')
444
- action_type = 'L'
445
 
446
 
447
  # Store the actions
 
221
 
222
 
223
  for action_type, pos in previous_actions: #[-8:]:
224
+ #print ('here3', action_type, pos)
225
  if action_type == 'move':
226
  action_type = 'N'
227
  if action_type == 'left_click':
 
439
 
440
  action_type = data.get("action_type")
441
  mouse_position = data.get("mouse_position")
442
+ if np.random.random() < 0.7:
443
  print ('setting left click')
444
+ action_type = 'left_click'
445
 
446
 
447
  # Store the actions