From dccf3cad404f1d54db61579af5354be4bd3dd133 Mon Sep 17 00:00:00 2001
From: Calvin <calvinpieters@gmail.com>
Date: Wed, 13 Dec 2023 13:42:08 +0200
Subject: [PATCH] Updated args['trsh'] if condition

.

.
---
 arc/scheduler.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/arc/scheduler.py b/arc/scheduler.py
index 0a8af14861..9e133767d4 100644
--- a/arc/scheduler.py
+++ b/arc/scheduler.py
@@ -784,7 +784,16 @@ def run_job(self,
             self.deduce_job_adapter(level=Level(repr=level_of_theory), job_type=job_type)
         args = {'keyword': {}, 'block': {}}
         if trsh:
-            args['trsh'] = {'trsh': trsh}
+            if isinstance(trsh, (str, list)):
+                args['trsh'] = {'trsh': trsh}
+            elif isinstance(trsh, dict) and 'trsh' in args:
+                for key, value in trsh.items():
+                    if isinstance(args['trsh'][key], list) and isinstance(value, list) and key in args['trsh']:
+                        args['trsh'][key].extend(value)
+                    else:
+                        args['trsh'][key] = value
+            else:
+                args['trsh'] = trsh
         if shift:
             args['shift'] = shift
         if scan_trsh: