Mercurial > repos > bimib > marea_2_0

changeset 119:ccd4d55699b5 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author luca_milaz
date Tue, 02 Jul 2024 20:21:11 +0000
parents bad889c7214b
children fcc11ca1e27e
files marea_2_0/flux_sampling.py
diffstat 1 files changed, 7 insertions(+), 6 deletions(-) [+]
line wrap: on
line diff
--- a/marea_2_0/flux_sampling.py	Tue Jul 02 20:21:00 2024 +0000
+++ b/marea_2_0/flux_sampling.py	Tue Jul 02 20:21:11 2024 +0000
@@ -37,7 +37,11 @@
     
     parser.add_argument('-in', '--input',
                         required = True,
-                        help = 'input model')
+                        help = 'inputs model')
+    
+    parser.add_argument('-in', '--model_name',
+                        required = True,
+                        help = 'model_names')
     
     parser.add_argument('-a', '--algorithm',
                         type = str,
@@ -182,15 +186,12 @@
     ARGS.output_folder = 'flux_sampling/'
 
 
-    modelNames = []
-    modelNames[0] = os.path.splitext(os.path.basename(ARGS.input_model[0]))[0] # extract model filename only
-    modelNames[1] = os.path.splitext(os.path.basename(ARGS.input_model[1]))[0] # extract model filename only
 
     # load custom model
     model1 = load_custom_model(
-        utils.FilePath.fromStrPath(ARGS.input_model[0]), modelNames[0].ext)
+        utils.FilePath.fromStrPath(ARGS.input[0]), ARGS.model_name[0].ext)
     model2 = load_custom_model(
-        utils.FilePath.fromStrPath(ARGS.input_model[1]), modelNames[1].ext)
+        utils.FilePath.fromStrPath(ARGS.input[1]), ARGS.model_name[1].ext)
 
     with open(ARGS.output_folder, 'w') as f:
                 f.write([model1, model2])