feat: display results in 'not found' error

src/sciline/pipeline.py

@@ -11,7 +11,7 @@
 from ._utils import key_name
 from .data_graph import DataGraph, to_task_graph
 from .display import pipeline_html_repr
-from .handler import ErrorHandler, HandleAsComputeTimeException
+from .handler import ErrorHandler, HandleAsComputeTimeException, UnsatisfiedRequirement
 from .scheduler import Scheduler
 from .task_graph import TaskGraph
 from .typing import Key
@@ -107,7 +107,7 @@ def visualize(
             Keyword arguments passed to :py:class:`graphviz.Digraph`.
         """
         if tp is None:
-            tp = self.leafs()
+            tp = self.final_result_keys()
         return self.get(tp, handler=HandleAsComputeTimeException()).visualize(**kwargs)
 
     def get(
@@ -140,7 +140,11 @@ def get(
             targets = tuple(keys)  # type: ignore[arg-type]
         else:
             targets = (keys,)  # type: ignore[assignment]
-        graph = to_task_graph(self, targets=targets, handler=handler)
+        try:
+            graph = to_task_graph(self, targets=targets, handler=handler)
+        except UnsatisfiedRequirement as e:
+            final_result_keys = ", ".join(map(repr, self.final_result_keys()))
+            raise type(e)(f'Did you meant one of: {final_result_keys}?') from e
         return TaskGraph(
             graph=graph,
             targets=targets if multi else keys,  # type: ignore[arg-type]
@@ -205,7 +209,7 @@ def _repr_html_(self) -> str:
         nodes = ((key, data) for key, data in self.underlying_graph.nodes.items())
         return pipeline_html_repr(nodes)
 
-    def leafs(self) -> tuple[type, ...]:
+    def final_result_keys(self) -> tuple[type, ...]:
         """Returns the keys that are not inputs to any other providers."""
         sink_nodes = [
             cast(type, node)

tests/pipeline_test.py

@@ -10,6 +10,7 @@
 import pytest
 
 import sciline as sl
+from sciline._utils import key_name
 
 
 def int_to_float(x: int) -> float:
@@ -1416,11 +1417,26 @@ def bad(x: int, y: int) -> float:
         pipeline.insert(bad)
 
 
-def test_leafs_method() -> None:
+def test_final_result_keys_method() -> None:
     def make_float() -> float:
         return 1.0
 
     def make_str(x: int) -> str:
         return "a string"
 
-    assert sl.Pipeline([make_float, make_str]).leafs() == (float, str)
+    assert sl.Pipeline([make_float, make_str]).final_result_keys() == (float, str)
+
+
+@pytest.mark.parametrize('get_method', ['get', 'compute'])
+def test_final_result_keys_in_not_found_error_message(get_method) -> None:
+    def make_float() -> float:
+        return 1.0
+
+    def make_str(x: int) -> str:
+        return "a string"
+
+    pl = sl.Pipeline([make_float, make_str])
+    with pytest.raises(sl.handler.UnsatisfiedRequirement) as info:
+        getattr(pl, get_method)(int)
+    for key in pl.final_result_keys():
+        assert key_name(key) in info.value.args[0]