FileNotFoundError fixes

jodisfields · Apr 22, 2023 · e98c0b7 · e98c0b7
1 parent 5b060c5
commit e98c0b7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/open_biomed/tasks/mol_task/dti.py b/open_biomed/tasks/mol_task/dti.py
@@ -244,6 +244,12 @@ def add_arguments(parser):
     parser = argparse.ArgumentParser()
     parser = add_arguments(parser)
     args = parser.parse_args()
+
+    output_dir = os.path.dirname(args.output_path)
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
 
     config = json.load(open(args.config_path, "r"))
-    main(args, config)
+    main(args, config)
+
+