Commit
·
7e0be42
1
Parent(s):
b48ceb9
Update app.py
Browse files
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-
|
156 |
if video.endswith(f'.{FORMAT}'):
|
157 |
if filter:
|
158 |
-
if len(video.split('_')) <
|
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]
|