diff --git a/arc/scheduler.py b/arc/scheduler.py index 9e133767d4..e6fe1ae784 100644 --- a/arc/scheduler.py +++ b/arc/scheduler.py @@ -735,7 +735,7 @@ def run_job(self, reactions: Optional[List['ARCReaction']] = None, scan_trsh: Optional[str] = '', shift: Optional[str] = '', - trsh: Optional[str] = '', + trsh: Optional[Union[str, dict, list]] = None, torsions: Optional[List[List[int]]] = None, times_rerun: int = 0, tsg: Optional[int] = None, @@ -1012,7 +1012,7 @@ def _run_a_job(self, max_job_time=job.max_job_time, rotor_index=job.rotor_index, reactions=job.reactions, - trsh=list(job.args['trsh'].values()) if 'trsh' in job.args else '', + trsh=job.args['trsh'] if 'trsh' in job.args else {}, torsions=job.torsions, times_rerun=job.times_rerun + int(rerun), tsg=job.tsg,