pytensor 3.0.6__tar.gz → 3.0.7__tar.gz

{pytensor-3.0.6/pytensor.egg-info → pytensor-3.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 3.0.6
+Version: 3.0.7
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License-Expression: BSD-3-Clause

{pytensor-3.0.6 → pytensor-3.0.7}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2026-06-15T11:44:50+0200",
+ "date": "2026-06-15T16:18:33+0200",
  "dirty": false,
  "error": null,
- "full-revisionid": "c141306ae1f92ce9049f37e0ea11af7d63089892",
- "version": "3.0.6"
+ "full-revisionid": "cb44be3432acef912fa74cb068620d011901f564",
+ "version": "3.0.7"
 }
 '''  # END VERSION_JSON
 

{pytensor-3.0.6 → pytensor-3.0.7}/pytensor/compile/builders.py

@@ -24,7 +24,7 @@ from pytensor.graph.basic import (
 from pytensor.graph.fg import FrozenFunctionGraph, FunctionGraph
 from pytensor.graph.null_type import NullType
 from pytensor.graph.op import HasInnerGraph, Op, io_connection_pattern
-from pytensor.graph.replace import clone_replace, graph_replace
+from pytensor.graph.replace import clone_replace
 from pytensor.graph.traversal import graph_inputs
 from pytensor.graph.utils import MissingInputError
 
@@ -899,7 +899,7 @@ class OpFromGraph(Op, HasInnerGraph):
 
         repl = dict(zip(self.inner_inputs, node.inputs, strict=True))
         clone_out_shapes = [s for s in out_shapes if isinstance(s, tuple)]
-        cloned = graph_replace(sum(clone_out_shapes, ()), replace=repl, strict=False)
+        cloned = clone_replace(sum(clone_out_shapes, ()), replace=repl)
         ret = []
         used = 0
         for i, out_shape in enumerate(out_shapes):

{pytensor-3.0.6 → pytensor-3.0.7}/pytensor/tensor/rewriting/indexed_elemwise.py

@@ -6,7 +6,6 @@ so the Numba backend can generate a single loop with indirect indexing,
 eliminating materialised intermediate arrays.
 """
 
-from pytensor.compile import optdb
 from pytensor.compile.builders import OpFromGraph
 from pytensor.graph import node_rewriter
 from pytensor.graph.rewriting.basic import GraphRewriter, dfs_rewriter
@@ -170,17 +169,23 @@ def undo_take_reshape_for_fusion(fgraph, node):
 
 
 indexed_elemwise_optdb = SequenceDB()
-optdb.register(
-    "fuse_indexed_into_elemwise",
-    indexed_elemwise_optdb,
-    "numba",
-    # symbolic_op_recognition is excluded from OpFromGraph inner-graph
-    # compilation, preventing recursive fusion.
-    "symbolic_op_recognition",
-    # After inplace_elemwise (position=50.5) so we see final inplace patterns,
-    # same position as other numba-specific rewrites (BlockwiseWithCoreShape).
-    position=100,
-)
+# Disabled by default on the 3.0.x line. This fusion shipped in 3.0.4 but can
+# trigger a RecursionError during OpFromGraph shape inference on some graphs
+# (e.g. numba-mode FullRankADVI); it was too risky for a minor release. The
+# proper fix lives on main (non-recursive core-shape construction). Until that
+# is backported we do not register the fusion into the default numba pipeline.
+# To re-enable, re-add `from pytensor.compile import optdb` and the call below.
+# optdb.register(
+#     "fuse_indexed_into_elemwise",
+#     indexed_elemwise_optdb,
+#     "numba",
+#     # symbolic_op_recognition is excluded from OpFromGraph inner-graph
+#     # compilation, preventing recursive fusion.
+#     "symbolic_op_recognition",
+#     # After inplace_elemwise (position=50.5) so we see final inplace patterns,
+#     # same position as other numba-specific rewrites (BlockwiseWithCoreShape).
+#     position=100,
+# )
 
 indexed_elemwise_optdb.register(
     "undo_take_dimshuffle_for_fusion",

{pytensor-3.0.6 → pytensor-3.0.7/pytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 3.0.6
+Version: 3.0.7
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License-Expression: BSD-3-Clause