File tree Expand file tree Collapse file tree 1 file changed +17
-5
lines changed
Expand file tree Collapse file tree 1 file changed +17
-5
lines changed Original file line number Diff line number Diff line change @@ -25,13 +25,13 @@ def main():
2525 args = parser .parse_args ()
2626
2727 # Check if the mode is valid
28- mode = "sequential"
29- if args .mode not in ["sequential" , "parallel" ]:
30- print (f"Mode { args .mode } not valid. Please use 'seq' or 'par'" )
28+ valid_modes = ["sequential" , "parallel" ]
29+ mode = args .mode if args .mode in valid_modes else "sequential"
30+ if args .mode and args .mode not in valid_modes :
31+ print (f"Mode { args .mode } not valid. Please use 'sequential' or 'parallel'" )
3132 print ("Setting sequential mode as default" )
3233 else :
33- print (f"Running in { args .mode } mode" )
34- mode = args .mode
34+ print (f"Running in { mode } mode" )
3535
3636 # put in a list all the elementes in the gen flag
3737 noConfGen = ["pythia8pp" , "pythia8hf" , "pythia8hi" , "pythia8powheg" ]
@@ -63,6 +63,18 @@ def main():
6363 "iniFile" : ""
6464 }
6565 })
66+ elif gen == "evtpool" :
67+ gens .append ({
68+ 'name' : 'evtpool' ,
69+ 'config' : {
70+ "eventPoolPath" : "/path/to/filename.root" ,
71+ "skipNonTrackable" : True ,
72+ "roundRobin" : False ,
73+ "randomize" : True ,
74+ "rngseed" : 0 ,
75+ "randomphi" : False ,
76+ }
77+ })
6678 elif gen == "extkinO2" :
6779 gens .append ({
6880 'name' : 'extkinO2' ,
You can’t perform that action at this time.
0 commit comments