# HG changeset patch
# User luca_milaz
# Date 1719951671 0
# Node ID ccd4d55699b5ff7053a9dd74e78a22e4a9774818
# Parent  bad889c7214bc261a1a4d08a0372b73db8338235
Uploaded

diff -r bad889c7214b -r ccd4d55699b5 marea_2_0/flux_sampling.py
--- 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])