ego/select.py

import os, shutil
import random
random.seed(43)

video_dir = "/scratch2/nlp/patrick/data/ZeroShotQA/EgoSchema_Zero_Shot_QA/videos"

all_files = os.listdir(video_dir)

videos = random.sample(all_files, 20)

for video in videos:
    shutil.copy(f"{video_dir}/{video}", f"/scratch/nlp/patrick/codes/VidLNeedleFinder/needlehaystack/haystack/ego/{video}")