Improve docstring for explicit_graph_inputs

Author: Dhruvanshu-Joshi

pytensor/graph/basic.py

@@ -940,37 +940,39 @@ def explicit_graph_inputs(
     graph: Variable[Any, Any] | Iterable[Variable[Any, Any]],
 ) -> Generator[Variable, None, None]:
     """
-    Get the inputs into PyTensor variables
+    Get the root variables needed as inputs to a function that computes `graph`
 
     Parameters
     ----------
-    graph: TensorVariable
-        Output `Variable` instances from which to search backward through
+    graph : TensorVariable
+        Output `Variable` instances for which to search backward through
         owners.
 
     Returns
     -------
-        Tensor variables that are input nodes with no owner, in the order
-        found by a left-recursive depth-first search started at the nodes in `graphs`.
+    iterable
+        Generator of root Variables (without owner) needed to compile a function that evaluates `graphs`.
 
     Examples
     --------
 
     .. code-block:: python
 
         import pytensor
+        import numpy as np
         import pytensor.tensor as pt
+        from pytensor.graph.basic import explicit_graph_inputs
 
         x = pt.vector('x')
         y = pt.constant(2)
         z = pt.mul(x*y)
 
-        pytensor.dprint(graph_inputs([z]))
-        # x [id A]
-        # 2 [id B]
+        out = list(explicit_graph_inputs(z))
+        f = pytensor.function([x], out)
+        eval = f(np.array([1, 2, 3]))
 
-        pytensor.dprint(explicit_graph_inputs([z]))
-        # x [id A]
+        print(eval)
+        # [array([1., 2., 3.])]
     """
     from pytensor.compile.sharedvalue import SharedVariable
 

tests/graph/test_basic.py

@@ -532,6 +532,7 @@ def test_explicit_graph_inputs():
 
     res = list(explicit_graph_inputs([a]))
     res1 = list(explicit_graph_inputs(b))
+
     assert res, res1 == [x]