quantumiracle-git commited on
Commit
d5bda0c
·
1 Parent(s): 1ed9279

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -203,6 +203,7 @@ def randomly_select_videos(env_name):
203
  with open(VIDEO_INFO, 'r') as fp:
204
  files = json.load(fp)
205
  env_files = files[env_name]
 
206
  # randomly choose two videos
207
  selected_video_ids = np.random.choice(len(env_files), 2, replace=False)
208
  left_video_path = env_files[selected_video_ids[0]]
 
203
  with open(VIDEO_INFO, 'r') as fp:
204
  files = json.load(fp)
205
  env_files = files[env_name]
206
+ print(env_name, len(env_files), env_files)
207
  # randomly choose two videos
208
  selected_video_ids = np.random.choice(len(env_files), 2, replace=False)
209
  left_video_path = env_files[selected_video_ids[0]]