Speedup graph traversal functions

doc/gallery/optimize/root.ipynb

@@ -1772,7 +1772,9 @@
     }
    ],
    "source": [
-    "from pytensor.graph.basic import explicit_graph_inputs\n",
+    "\n",
+    "from pytensor.graph.traversal import explicit_graph_inputs\n",
+    "\n",
     "list(explicit_graph_inputs(w_bar_2))"
    ]
   },

pytensor/compile/builders.py

@@ -16,13 +16,12 @@
     Constant,
     NominalVariable,
     Variable,
-    graph_inputs,
-    io_connection_pattern,
 )
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.null_type import NullType
-from pytensor.graph.op import HasInnerGraph, Op
+from pytensor.graph.op import HasInnerGraph, Op, io_connection_pattern
 from pytensor.graph.replace import clone_replace
+from pytensor.graph.traversal import graph_inputs
 from pytensor.graph.utils import MissingInputError
 
 

pytensor/compile/debugmode.py

@@ -27,11 +27,12 @@
 from pytensor.compile.mode import Mode, register_mode
 from pytensor.compile.ops import OutputGuard, _output_guard
 from pytensor.configdefaults import config
-from pytensor.graph.basic import Variable, io_toposort
+from pytensor.graph.basic import Variable
 from pytensor.graph.destroyhandler import DestroyHandler
 from pytensor.graph.features import AlreadyThere, BadOptimization
 from pytensor.graph.fg import Output
 from pytensor.graph.op import HasInnerGraph, Op
+from pytensor.graph.traversal import io_toposort
 from pytensor.graph.utils import InconsistencyError, MethodNotDefined
 from pytensor.link.basic import Container, LocalLinker
 from pytensor.link.c.op import COp

pytensor/compile/function/types.py

@@ -20,14 +20,13 @@
 from pytensor.graph.basic import (
     Constant,
     Variable,
-    ancestors,
     clone_get_equiv,
-    graph_inputs,
 )
 from pytensor.graph.destroyhandler import DestroyHandler
 from pytensor.graph.features import AlreadyThere, Feature, PreserveVariableAttributes
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.op import HasInnerGraph
+from pytensor.graph.traversal import ancestors, graph_inputs
 from pytensor.graph.utils import InconsistencyError, get_variable_trace_string
 from pytensor.link.basic import Container
 from pytensor.link.utils import raise_with_op

pytensor/d3viz/formatting.py

@@ -10,8 +10,9 @@
 
 import pytensor
 from pytensor.compile import Function, builders
-from pytensor.graph.basic import Apply, Constant, Variable, graph_inputs
+from pytensor.graph.basic import Apply, Constant, Variable
 from pytensor.graph.fg import FunctionGraph
+from pytensor.graph.traversal import graph_inputs
 from pytensor.printing import _try_pydot_import
 
 

pytensor/graph/__init__.py

@@ -5,10 +5,9 @@
     Apply,
     Variable,
     Constant,
-    graph_inputs,
     clone,
-    ancestors,
 )
+from pytensor.graph.traversal import ancestors, graph_inputs
 from pytensor.graph.replace import clone_replace, graph_replace, vectorize_graph
 from pytensor.graph.op import Op
 from pytensor.graph.type import Type