Fix info (#590)

* Fix info

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

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

* fix type check

* Update tests/test_dependencies_executor.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

* Update test_dependencies_executor.py

* Update dependency.py

* Update dependency.py

---------

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

Author: 3 people

executorlib/interactive/dependency.py

@@ -65,6 +65,23 @@ def __init__(
         else:
             self._generate_dependency_graph = True
 
+    @property
+    def info(self) -> Optional[dict]:
+        """
+        Get the information about the executor.
+
+        Returns:
+            Optional[dict]: Information about the executor.
+        """
+        if isinstance(self._future_queue, queue.Queue):
+            f: Future = Future()
+            self._future_queue.queue.insert(
+                0, {"internal": True, "task": "info", "future": f}
+            )
+            return f.result()
+        else:
+            return None
+
     def submit(  # type: ignore
         self,
         fn: Callable[..., Any],
@@ -168,6 +185,11 @@ def _execute_tasks_with_dependencies(
             future_queue.task_done()
             future_queue.join()
             break
+        if (  # shutdown the executor
+            task_dict is not None and "internal" in task_dict and task_dict["internal"]
+        ):
+            if task_dict["task"] == "info":
+                task_dict["future"].set_result(executor.info)
         elif (  # handle function submitted to the executor
             task_dict is not None and "fn" in task_dict and "future" in task_dict
         ):

tests/test_dependencies_executor.py

@@ -8,6 +8,7 @@
 from executorlib.interfaces.single import create_single_node_executor
 from executorlib.interactive.dependency import _execute_tasks_with_dependencies
 from executorlib.standalone.serialize import cloudpickle_register
+from executorlib.standalone.interactive.spawner import MpiExecSpawner
 
 
 try:
@@ -327,3 +328,36 @@ def test_block_allocation_true_two_workers_loop(self):
                         parameter=lst,
                     )
                 lst.result()
+
+
+class TestInfo(unittest.TestCase):
+    """Test cases for the info property of SingleNodeExecutor."""
+
+    def setUp(self):
+        """Set up the expected info dictionary."""
+        self.expected_info = {
+            'cores': 1,
+            'cwd': None,
+            'openmpi_oversubscribe': False,
+            'cache_directory': None,
+            'hostname_localhost': None,
+            'spawner': MpiExecSpawner,
+            'max_cores': None,
+            'max_workers': None,
+        }
+
+    def test_info_disable_dependencies_true(self):
+        """Test info property with dependencies disabled."""
+        with SingleNodeExecutor(disable_dependencies=True) as exe:
+            self.assertEqual(exe.info, self.expected_info)
+
+    def test_info_disable_dependencies_false(self):
+        """Test info property with dependencies enabled."""
+        with SingleNodeExecutor(disable_dependencies=False) as exe:
+            self.assertEqual(exe.info, self.expected_info)
+
+    def test_info_error_handling(self):
+        """Test info property error handling when executor is not running."""
+        exe = SingleNodeExecutor()
+        exe.shutdown(wait=True)
+        self.assertIsNone(exe.info)