store changes for graph

Author: Dhruvanshu-Joshi

pytensor/graph/basic.py

@@ -907,7 +907,8 @@ def graph_inputs(
     yield from (r for r in ancestors(graphs, blockers) if r.owner is None)
 
 
-def required_graph_inputs(graph):
+# def required_graph_inputs(graph:Variable | Iterator[Variable])-> Generator[Variable, None, None]:
+def required_graph_inputs(graph: list[Variable[Any, Any]]) -> list[Variable[Any, Any]]:
     """
     Get the inputs into PyTensor variables
 
@@ -921,13 +922,38 @@ def required_graph_inputs(graph):
     -------
         List of 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`.
+
+    Examples
+    --------
+    >>> import pytensor as pt
+    >>> x=pt.vector('x')
+    >>> y=pt.constant('y')
+    >>> z = pt.mul(x*y)
+    >>> required_graph_inputs([a])
+    [[[ 0  1  2  3]
+      [ 4  5  6  7]
+      [ 8  9 10 11]]
+
+     [[12 13 14 15]
+      [16 17 18 19]
+      [20 21 22 23]]]
+
+
+    >>> pt.matrix_transpose(x).eval()
+    [[[ 0  4  8]
+      [ 1  5  9]
+      [ 2  6 10]
+      [ 3  7 11]]
+
+     [[12 16 20]
+      [13 17 21]
+      [14 18 22]
+      [15 19 23]]]
     """
     from pytensor.compile.sharedvalue import SharedVariable
 
-    if isinstance(graph, tuple | list):
-        graph = graph
-    else:
-        graph = [graph]
+    # if isinstance(graph, Variable):
+    #     graph = [graph]
 
     return [
         v

tests/graph/test_basic.py

@@ -8,7 +8,6 @@
 from pytensor import tensor as pt
 from pytensor.graph.basic import (
     Apply,
-    Constant,
     NominalVariable,
     Variable,
     ancestors,
@@ -63,16 +62,6 @@ def MyVariable(thingy):
     return Variable(MyType(thingy), None, None)
 
 
-def MyConstant(name, thingy, data=None):
-    return Constant(MyType(thingy), data, name=name)
-
-
-# def MyConstant(thingy):
-#     return Constant(MyType(thingy), None, None)
-# def MyVariable(thingy):
-#     return Variable(MyType(thingy), None, None)
-
-
 class MyOp(Op):
     __props__ = ()
 
@@ -87,20 +76,6 @@ def perform(self, *args, **kwargs):
         raise NotImplementedError("No Python implementation available.")
 
 
-class MyCustomOp(Op):
-    __props__ = ()
-
-    def make_node(self, *inputs):
-        for input in inputs:
-            assert isinstance(input, Variable)
-            assert isinstance(input.type, MyType)
-        outputs = [MyVariable(sum(input.type.thingy for input in inputs))]
-        return Apply(self, list(inputs), outputs)
-
-    def perform(self, *args, **kwargs):
-        raise NotImplementedError("No Python implementation available.")
-
-
 MyOp = MyOp()