added condition to not do gitkeep

Zeyu-Li · Jan 19, 2021 · 4ef6e1f · 4ef6e1f
1 parent f30c59f
commit 4ef6e1f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/split.py b/split.py
@@ -26,8 +26,11 @@ def main():
     dir_list = os.listdir(inputs)
     # only splits first file
     input_file = dir_list[0]
+    if input_file == '.gitkeep':
+        input_file = dir_list[1]
 
     # tests
+    # print(input_file)
     # print(sys.argv)
 
     splits = []