[feature] Add support for future objects in input dictionaries

Author: jan-janssen

executorlib/interactive/shared.py

@@ -483,6 +483,8 @@ def get_result(arg: Union[list[Future], Future]) -> Any:
             return arg.result()
         elif isinstance(arg, list):
             return [get_result(arg=el) for el in arg]
+        elif isinstance(arg, dict):
+            return {k: get_result(arg=v) for k, v in arg.items()}
         else:
             return arg
 
@@ -510,6 +512,8 @@ def find_future_in_list(lst):
                 future_lst.append(el)
             elif isinstance(el, list):
                 find_future_in_list(lst=el)
+            elif isinstance(el, dict):
+                find_future_in_list(lst=el.values())
 
     find_future_in_list(lst=task_dict["args"])
     find_future_in_list(lst=task_dict["kwargs"].values())

tests/test_dependencies_executor.py

@@ -38,6 +38,10 @@ def merge(lst):
     return sum(lst)
 
 
+def return_input_dict(input_dict):
+    return input_dict
+
+
 def raise_error():
     raise RuntimeError
 
@@ -130,6 +134,14 @@ def test_many_to_one(self):
             )
             self.assertEqual(future_sum.result(), 15)
 
+    def test_future_input_dict(self):
+        with SingleNodeExecutor() as exe:
+            fs = exe.submit(
+                return_input_dict,
+                input_dict={"a": exe.submit(sum, [2, 2])},
+            )
+            self.assertEqual(fs.result()["a"], 4)
+
 
 class TestExecutorErrors(unittest.TestCase):
     def test_block_allocation_false_one_worker(self):