1 files changed, 7 insertions, 0 deletions

diff --git a/pyecsca/misc/utils.py b/pyecsca/misc/utils.py
index e525fe0..f19009e 100644
--- a/pyecsca/misc/utils.py
+++ b/pyecsca/misc/utils.py
@@ -10,10 +10,12 @@ from concurrent.futures import ProcessPoolExecutor, as_completed, Future
 
 
 def pexec(s):
+    """Parse with exec."""
     return parse(s, mode="exec")
 
 
 def peval(s):
+    """Parse with eval."""
     return parse(s, mode="eval")
 
 
@@ -55,7 +57,9 @@ class TaskExecutor(ProcessPoolExecutor):
     """A simple ProcessPoolExecutor that keeps tracks of tasks that were submitted to it."""
 
     keys: List[Any]
+    """A list of keys that identify the futures."""
     futures: List[Future]
+    """A list of futures submitted to the executor."""
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -63,6 +67,7 @@ class TaskExecutor(ProcessPoolExecutor):
         self.futures = []
 
     def submit_task(self, key: Any, fn, /, *args, **kwargs):
+        """Submit a task (function `fn`), identified by `key` and with `args` and `kwargs`."""
         future = self.submit(fn, *args, **kwargs)
         self.futures.append(future)
         self.keys.append(key)
@@ -70,9 +75,11 @@ class TaskExecutor(ProcessPoolExecutor):
 
     @property
     def tasks(self):
+        """A list of tasks that were submitted to this executor."""
         return list(zip(self.keys, self.futures))
 
     def as_completed(self) -> Generator[tuple[Any, Future], Any, None]:
+        """Like `concurrent.futures.as_completed`, but yields a pair of key and future."""
         for future in as_completed(self.futures):
             i = self.futures.index(future)
             yield self.keys[i], future