diff --git a/monoloco/eval/generate_kitti.py b/monoloco/eval/generate_kitti.py index f65021e..5da2843 100644 --- a/monoloco/eval/generate_kitti.py +++ b/monoloco/eval/generate_kitti.py @@ -273,5 +273,5 @@ def create_empty_files(dir_out, net): for i in range(7481): name = "0" * (6 - len(str(i))) + str(i) + '.txt' - ff = open(os.path.join(dir_out[net], name), "a+") - ff.close() + with open(os.path.join(dir_out[net], name), "a+"): + pass diff --git a/monoloco/utils/kitti.py b/monoloco/utils/kitti.py index a306ff1..5192500 100644 --- a/monoloco/utils/kitti.py +++ b/monoloco/utils/kitti.py @@ -161,8 +161,8 @@ def read_and_rewrite(path_orig, path_new): line_new = temp_1 + ' ' + hwl + ' ' + temp_2 + '\n' ff.write("%s" % line_new) except FileNotFoundError: - ff = open(path_new, "a+") - ff.close() + with open(path_new, "a+"): + pass def find_cluster(dd, clusters):