Plot: Invert the future hash dictionary inside the function (#822)

* Plot: Invert the future hash dictionary inside the function

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: jan-janssen

executorlib/standalone/plot.py

@@ -81,15 +81,13 @@ def add_element(arg, link_to, label=""):
     return node_lst, edge_lst
 
 
-def generate_task_hash_for_plotting(
-    task_dict: dict, future_hash_inverse_dict: dict
-) -> bytes:
+def generate_task_hash_for_plotting(task_dict: dict, future_hash_dict: dict) -> bytes:
     """
     Generate a hash for a task dictionary.
 
     Args:
         task_dict (dict): Dictionary containing task information.
-        future_hash_inverse_dict (dict): Dictionary mapping future hash to future object.
+        future_hash_dict (dict): Dictionary mapping future hash to future object.
 
     Returns:
         bytes: Hash generated for the task dictionary.
@@ -121,12 +119,13 @@ def convert_arg(arg, future_hash_inverse_dict):
         else:
             return arg
 
+    future_hash_inverted_dict = {v: k for k, v in future_hash_dict.items()}
     args_for_hash = [
-        convert_arg(arg=arg, future_hash_inverse_dict=future_hash_inverse_dict)
+        convert_arg(arg=arg, future_hash_inverse_dict=future_hash_inverted_dict)
         for arg in task_dict["args"]
     ]
     kwargs_for_hash = {
-        k: convert_arg(arg=v, future_hash_inverse_dict=future_hash_inverse_dict)
+        k: convert_arg(arg=v, future_hash_inverse_dict=future_hash_inverted_dict)
         for k, v in task_dict["kwargs"].items()
     }
     return cloudpickle.dumps(

executorlib/task_scheduler/interactive/dependency.py

@@ -146,9 +146,7 @@ def submit(  # type: ignore
             }
             task_hash = generate_task_hash_for_plotting(
                 task_dict=task_dict,
-                future_hash_inverse_dict={
-                    v: k for k, v in self._future_hash_dict.items()
-                },
+                future_hash_dict=self._future_hash_dict,
             )
             self._future_hash_dict[task_hash] = f
             self._task_hash_dict[task_hash] = task_dict