shutil for rm

This commit is contained in:
Max Hoffman
2023-02-27 13:11:27 -08:00
parent 81bc6e6e80
commit 07c9fa813c

View File

@@ -1,4 +1,5 @@
import os
import shutil
import sys
import random
@@ -14,9 +15,7 @@ adds = int(sys.argv[4])
if __name__=="__main__":
if not os.path.exists(table_dir):
try:
os.rmdir(table_dir)
except:
shutil.rmtree(table_dir, ignore_errors=True)
os.makedirs(table_dir)
ys = [i for i in range(rows+adds)]