yuntian-deng commited on
Commit
e17298c
·
1 Parent(s): ab84bf5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -209,6 +209,8 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
209
  print ('here3', action_type, pos)
210
  if action_type == 'move':
211
  action_type = 'N'
 
 
212
  if action_type == "N":
213
  x, y = pos
214
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
 
209
  print ('here3', action_type, pos)
210
  if action_type == 'move':
211
  action_type = 'N'
212
+ if action_type == 'left_click':
213
+ action_type = 'L'
214
  if action_type == "N":
215
  x, y = pos
216
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2