From 0a989989a3f0826d9eecc8b818e9769f93483189 Mon Sep 17 00:00:00 2001
From: Zen <46526140+master-of-zen@users.noreply.github.com>
Date: Sat, 11 Jan 2020 03:19:01 +0200
Subject: [PATCH] Update main.py

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 130a8c6..25ce8b3 100644
--- a/main.py
+++ b/main.py
@@ -106,7 +106,7 @@ def main(input_video, encoding_params, num_worker):
 
     # Make temporal directories, and remove them if already presented
     if os.path.isdir(join(os.getcwd(), "temp")):
-        shutil.rmtree(join(os.getcwd(), "temp"))
+        rmtree(join(os.getcwd(), "temp"))
 
     os.makedirs(join(os.getcwd(), 'temp', 'split'))
     os.makedirs(join(os.getcwd(), 'temp', 'encode'))
@@ -142,4 +142,4 @@ if __name__ == '__main__':
     print(f'Completed in {round(time.time()-start, 1)} seconds')
 
     # Delete temp folders
-    shutil.rmtree(join(os.getcwd(), "temp"))
+    rmtree(join(os.getcwd(), "temp"))