Skip to content

Commit

Permalink
Merge pull request #411 from mabruzzo/tweak-concat.py
Browse files Browse the repository at this point in the history
`concat.py` now ensures outdir exists
  • Loading branch information
evaneschneider authored Jan 30, 2025
2 parents 34d1a86 + aa2cdee commit cbd1c66
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion python_scripts/concat.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import argparse
import datetime
from functools import partial
import os
import pathlib

import concat_internals
Expand Down Expand Up @@ -126,6 +127,16 @@ def main(args):

#raise RuntimeError(repr(command_triples))

# create the output directory if it doesn't already exist...
abs_out_dir = os.path.abspath(args.output_directory)
if not os.path.exists(abs_out_dir):
if os.path.exists(os.path.dirname(abs_out_dir)):
os.mkdir(abs_out_dir)
else:
raise RuntimeError(
f"Can't create {args.output_directory} since the "
f"{args.output_directory}/.. directory doesn't already exist")

for output in args.concat_outputs:
print(f"concatenating {output}")
for dset_kind, build_source_path, concat_fn in command_triples:
Expand All @@ -144,4 +155,4 @@ def main(args):


if __name__ == '__main__':
main(parser.parse_args())
main(parser.parse_args())

0 comments on commit cbd1c66

Please sign in to comment.