pytensor 2.26.0__tar.gz → 2.26.2__tar.gz

{pytensor-2.26.0/pytensor.egg-info → pytensor-2.26.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytensor
-Version: 2.26.0
+Version: 2.26.2
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license:

{pytensor-2.26.0 → pytensor-2.26.2}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-11-05T21:53:01+0100",
+ "date": "2024-11-11T14:51:54+0100",
  "dirty": false,
  "error": null,
- "full-revisionid": "a570dbfd2cb8fd081f7c8074d7ce1c4d4b8c4469",
- "version": "2.26.0"
+ "full-revisionid": "fdbf3aa56a2f2e0e1bb8dc97816c8d0ba34ca5ed",
+ "version": "2.26.2"
 }
 '''  # END VERSION_JSON
 

{pytensor-2.26.0 → pytensor-2.26.2}/pytensor/graph/basic.py

@@ -622,9 +622,9 @@ class Variable(Node, Generic[_TypeType, OptionalApplyType]):
                 if isinstance(key, str):
                     matching_vars = get_var_by_name([self], key)
                     if not matching_vars:
-                        raise Exception(f"{key} not found in graph")
+                        raise ValueError(f"{key} not found in graph")
                     elif len(matching_vars) > 1:
-                        raise Exception(f"Found multiple variables with name {key}")
+                        raise ValueError(f"Found multiple variables with name {key}")
                     new_input_to_values[matching_vars[0]] = value
                 else:
                     new_input_to_values[key] = value

{pytensor-2.26.0 → pytensor-2.26.2}/pytensor/tensor/rewriting/basic.py

@@ -32,6 +32,7 @@ from pytensor.compile.ops import ViewOp
 from pytensor.graph import FunctionGraph
 from pytensor.graph.basic import Constant, Variable
 from pytensor.graph.rewriting.basic import (
+    NodeProcessingGraphRewriter,
     NodeRewriter,
     RemovalNodeRewriter,
     Rewriter,
@@ -1101,10 +1102,7 @@ def local_useless_split(fgraph, node):
 
 
 @node_rewriter(None)
-def constant_folding(fgraph, node):
-    if not node.op.do_constant_folding(fgraph, node):
-        return False
-
+def unconditional_constant_folding(fgraph, node):
     if not all(isinstance(inp, Constant) for inp in node.inputs):
         return False
 
@@ -1151,6 +1149,23 @@ def constant_folding(fgraph, node):
     return rval
 
 
+topo_unconditional_constant_folding = in2out(
+    unconditional_constant_folding,
+    ignore_newtrees=True,
+    name="topo_unconditional_constant_folding",
+    # Not all Ops have a perform method, so we ignore failures to constant_fold
+    failure_callback=NodeProcessingGraphRewriter.warn_ignore,
+)
+
+
+@node_rewriter(None)
+def constant_folding(fgraph, node):
+    if not node.op.do_constant_folding(fgraph, node):
+        return False
+
+    return unconditional_constant_folding.transform(fgraph, node)
+
+
 topo_constant_folding = in2out(
     constant_folding, ignore_newtrees=True, name="topo_constant_folding"
 )

{pytensor-2.26.0 → pytensor-2.26.2}/pytensor/tensor/rewriting/math.py

@@ -699,7 +699,10 @@ def local_div_switch_sink(fgraph, node):
             # will point to the new division op.
             copy_stack_trace(node.outputs, fdiv)
 
-            fct = switch(switch_cond, zero_switch_input, fdiv)
+            if branch == 0:
+                fct = switch(switch_cond, zero_switch_input, fdiv)
+            else:
+                fct = switch(switch_cond, fdiv, zero_switch_input)
 
             # Tell debug_mode than the output is correct, even if nan disappear
             fct.tag.values_eq_approx = values_eq_approx_remove_nan

{pytensor-2.26.0 → pytensor-2.26.2/pytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytensor
-Version: 2.26.0
+Version: 2.26.2
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license: