Skip to content

Commit f3587fd

Browse files
committed
fixed regex in parse_datasets to work with filenames not paths
1 parent 87765a2 commit f3587fd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

parse-data.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -110,11 +110,11 @@ def main():
110110
trial_id_to_youtube_logs = dict()
111111
trial_id_to_stall_logs = dict()
112112

113-
for f, trial_id in directory_traversal_helper.get_files_matching_regex(youtube_logs_directory, "[/]stall-log-(.+).txt"):
113+
for f, trial_id in directory_traversal_helper.get_files_matching_regex(youtube_logs_directory, "stall-log-(.+).txt"):
114114
assert(trial_id not in trial_id_to_stall_logs)
115115
trial_id_to_stall_logs[trial_id] = f
116116

117-
for f, trial_id in directory_traversal_helper.get_files_matching_regex(youtube_logs_directory, "[/]log-(.+).txt"):
117+
for f, trial_id in directory_traversal_helper.get_files_matching_regex(youtube_logs_directory, "^log-(.+).txt"):
118118
assert(trial_id not in trial_id_to_youtube_logs)
119119
trial_id_to_youtube_logs[trial_id] = f
120120

0 commit comments

Comments
 (0)