quantumiracle-git commited on
Commit
7e0be42
·
1 Parent(s): b48ceb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -152,10 +152,10 @@ def parse_envs(folder=VIDEO_PATH, filter=True, MAX_ITER=20000, DEFAULT_ITER=2000
152
  if os.path.isdir(env_path):
153
  videos = os.listdir(env_path)
154
  video_files = []
155
- for video in videos: # video name rule: EnvName_Alg_Seed_Timestamp_Checkpoint_video-episode_EpisodeID
156
  if video.endswith(f'.{FORMAT}'):
157
  if filter:
158
- if len(video.split('_')) < 7:
159
  print(f'{video} is wrongly named.')
160
  seed = video.split('_')[2]
161
  checkpoint = video.split('_')[4]
 
152
  if os.path.isdir(env_path):
153
  videos = os.listdir(env_path)
154
  video_files = []
155
+ for video in videos: # video name rule: EnvName_Alg_Seed_Timestamp_Checkpoint_video-episode-EpisodeID
156
  if video.endswith(f'.{FORMAT}'):
157
  if filter:
158
+ if len(video.split('_')) < 6:
159
  print(f'{video} is wrongly named.')
160
  seed = video.split('_')[2]
161
  checkpoint = video.split('_')[4]