From d260d7c0558324879bc4a4ef03ee2ff3d6c427e3 Mon Sep 17 00:00:00 2001 From: Vidya Koesmahargyo Date: Tue, 1 Dec 2020 09:55:23 -0500 Subject: [PATCH] fixed libraries --- process_data.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/process_data.py b/process_data.py index cd5e4e1c..116ef1b0 100644 --- a/process_data.py +++ b/process_data.py @@ -15,6 +15,8 @@ import argparse import logging import glob import time +import subprocess +from os.path import splitext logging.basicConfig(level=logging.INFO) logger=logging.getLogger() @@ -187,7 +189,7 @@ if __name__=="__main__": #add check for mov and mp3 here input_type = 'file' - if file_ext.lower() in ['.mp4','mov']: + if file_ext.lower() in ['.mp4','.mov']: if file_ext.lower() == '.mov': convert_file(args.input_path) process_raw_video_file(args, s_config, r_config)