pytensor 2.30.2__tar.gz → 2.31.0__tar.gz

{pytensor-2.30.2/pytensor.egg-info → pytensor-2.31.0}/PKG-INFO

@@ -1,53 +1,9 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 2.30.2
+Version: 2.31.0
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
-License: .. _license:
-        
-        LICENSE
-        =======
-        
-        Copyright (c) 2008-2019, Theano Development Team
-        Copyright (c) 2020-2021, PyMC Development team
-        Copyright (c) 2021-2022, Aesara Development Team
-        Copyright (c) 2022, PyMC Development team
-        All rights reserved.
-        
-        Contains code from NumPy, Copyright (c) 2005-2016, NumPy Developers.
-        Contains code from Aesara, Copyright (c) 2021-2022, Aesara Developers.
-        All rights reserved.
-        
-        theano/scan/*.py[c]: Razvan Pascanu, Frederic Bastien, James Bergstra, Pascal Lamblin, Arnaud Bergeron, PyMC Developers, PyTensor Developers, (c) 2010, Universite de Montreal
-        theano/tensor/sharedvar.py: James Bergstra, (c) 2010, Universite de Montreal, 3-clause BSD License
-        theano/gradient.py: James Bergstra, Razvan Pascanu, Arnaud Bergeron, Ian Goodfellow, PyMC Developers, PyTensor Developers, (c) 2011, Universite de Montreal, 3-clause BSD License
-        theano/compile/monitormode.py: this code was initially copied from the 'pyutools' package by its original author, and re-licensed under Theano's license.
-        
-        Contains frozendict code from slezica’s python-frozendict(https://github.com/slezica/python-frozendict/blob/master/frozendict/__init__.py), Copyright (c) 2012 Santiago Lezica. All rights reserved.
-        
-        Redistribution and use in source and binary forms, with or without
-        modification, are permitted provided that the following conditions are met:
-        
-            * Redistributions of source code must retain the above copyright
-              notice, this list of conditions and the following disclaimer.
-            * Redistributions in binary form must reproduce the above copyright
-              notice, this list of conditions and the following disclaimer in the
-              documentation and/or other materials provided with the distribution.
-            * Neither the name of PyTensor, Theano, nor the names of its contributors may be
-              used to endorse or promote products derived from this software without
-              specific prior written permission.
-        
-        THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''AS IS'' AND ANY
-        EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
-        WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
-        DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
-        DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
-        (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
-        LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
-        ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
-        (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
-        SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-        
+License-Expression: BSD-3-Clause
 Project-URL: homepage, https://github.com/pymc-devs/pytensor
 Project-URL: repository, https://github.com/pymc-devs/pytensor
 Project-URL: documentation, https://pytensor.readthedocs.io/en/latest/
@@ -56,7 +12,6 @@ Classifier: Development Status :: 6 - Mature
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Science/Research
 Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: BSD License
 Classifier: Programming Language :: Python
 Classifier: Topic :: Software Development :: Code Generators
 Classifier: Topic :: Software Development :: Compilers

{pytensor-2.30.2 → pytensor-2.31.0}/doc/extending/extending_pytensor_solution_1.py

@@ -118,7 +118,7 @@ class TestSumDiffOp(utt.InferShapeTester):
         self.op_class = SumDiffOp
 
     def test_perform(self):
-        rng = np.random.RandomState(43)
+        rng = np.random.default_rng(43)
         x = matrix()
         y = matrix()
         f = pytensor.function([x, y], self.op_class()(x, y))
@@ -128,7 +128,7 @@ class TestSumDiffOp(utt.InferShapeTester):
         assert np.allclose([x_val + y_val, x_val - y_val], out)
 
     def test_gradient(self):
-        rng = np.random.RandomState(43)
+        rng = np.random.default_rng(43)
 
         def output_0(x, y):
             return self.op_class()(x, y)[0]
@@ -150,7 +150,7 @@ class TestSumDiffOp(utt.InferShapeTester):
         )
 
     def test_infer_shape(self):
-        rng = np.random.RandomState(43)
+        rng = np.random.default_rng(43)
 
         x = dmatrix()
         y = dmatrix()

{pytensor-2.30.2 → pytensor-2.31.0}/doc/library/d3viz/index.ipynb

@@ -95,7 +95,7 @@
     "noutputs = 10\n",
     "nhiddens = 50\n",
     "\n",
-    "rng = np.random.RandomState(0)\n",
+    "rng = np.random.default_rng(0)\n",
     "x = pt.dmatrix('x')\n",
     "wh = pytensor.shared(rng.normal(0, 1, (nfeatures, nhiddens)), borrow=True)\n",
     "bh = pytensor.shared(np.zeros(nhiddens), borrow=True)\n",

{pytensor-2.30.2 → pytensor-2.31.0}/doc/library/d3viz/index.rst

@@ -58,7 +58,7 @@ hidden layer and a softmax output layer.
     noutputs = 10
     nhiddens = 50
 
-    rng = np.random.RandomState(0)
+    rng = np.random.default_rng(0)
     x = pt.dmatrix('x')
     wh = pytensor.shared(rng.normal(0, 1, (nfeatures, nhiddens)), borrow=True)
     bh = pytensor.shared(np.zeros(nhiddens), borrow=True)

{pytensor-2.30.2 → pytensor-2.31.0}/doc/optimizations.rst

@@ -239,7 +239,7 @@ Optimization                                              o4             o3  o2
         See :func:`insert_inplace_optimizer`
 
     inplace_random
-        Typically when a graph uses random numbers, the RandomState is stored
+        Typically when a graph uses random numbers, the random Generator is stored
         in a shared variable, used once per call and, updated after each function
         call.  In this common case, it makes sense to update the random number generator in-place.
 

{pytensor-2.30.2 → pytensor-2.31.0}/pyproject.toml

@@ -14,13 +14,13 @@ requires-python = ">=3.10,<3.14"
 authors = [{ name = "pymc-devs", email = "pymc.devs@gmail.com" }]
 description = "Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs."
 readme = "README.rst"
-license = { file = "LICENSE.txt" }
+license = "BSD-3-Clause"
+license-files = ["LICENSE.txt"]
 classifiers = [
     "Development Status :: 6 - Mature",
     "Intended Audience :: Education",
     "Intended Audience :: Science/Research",
     "Intended Audience :: Developers",
-    "License :: OSI Approved :: BSD License",
     "Programming Language :: Python",
     "Topic :: Software Development :: Code Generators",
     "Topic :: Software Development :: Compilers",
@@ -130,7 +130,7 @@ exclude = ["doc/", "pytensor/_version.py"]
 docstring-code-format = true
 
 [tool.ruff.lint]
-select = ["B905", "C", "E", "F", "I", "UP", "W", "RUF", "PERF", "PTH", "ISC", "T20", "NPY201"]
+select = ["C", "E", "F", "I", "UP", "W", "RUF", "PERF", "PTH", "ISC", "T20", "NPY201"]
 ignore = ["C408", "C901", "E501", "E741", "RUF012", "PERF203", "ISC001"]
 unfixable = [
     # zip-strict: the auto-fix adds `strict=False` but we might want `strict=True` instead

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-04-01T22:32:44+0200",
+ "date": "2025-05-19T16:51:53+0200",
  "dirty": false,
  "error": null,
- "full-revisionid": "0f5da80c16d5d053f39204d1a728e70e194bef6c",
- "version": "2.30.2"
+ "full-revisionid": "5ffe17ad7e1d2bc0841b69d0b57cfc9e142771c6",
+ "version": "2.31.0"
 }
 '''  # END VERSION_JSON
 

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/compile/builders.py

@@ -873,7 +873,6 @@ class OpFromGraph(Op, HasInnerGraph):
 
     def perform(self, node, inputs, outputs):
         variables = self.fn(*inputs)
-        assert len(variables) == len(outputs)
-        # strict=False because asserted above
-        for output, variable in zip(outputs, variables, strict=False):
+        # zip strict not specified because we are in a hot loop
+        for output, variable in zip(outputs, variables):
             output[0] = variable

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/compile/function/types.py

@@ -924,7 +924,8 @@ class Function:
 
         # Reinitialize each container's 'provided' counter
         if trust_input:
-            for arg_container, arg in zip(input_storage, args, strict=False):
+            # zip strict not specified because we are in a hot loop
+            for arg_container, arg in zip(input_storage, args):
                 arg_container.storage[0] = arg
         else:
             for arg_container in input_storage:
@@ -934,7 +935,8 @@ class Function:
                 raise TypeError("Too many parameter passed to pytensor function")
 
             # Set positional arguments
-            for arg_container, arg in zip(input_storage, args, strict=False):
+            # zip strict not specified because we are in a hot loop
+            for arg_container, arg in zip(input_storage, args):
                 # See discussion about None as input
                 # https://groups.google.com/group/theano-dev/browse_thread/thread/920a5e904e8a8525/4f1b311a28fc27e5
                 if arg is None:

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/compile/monitormode.py

@@ -104,10 +104,7 @@ def detect_nan(fgraph, i, node, fn):
     from pytensor.printing import debugprint
 
     for output in fn.outputs:
-        if (
-            not isinstance(output[0], np.random.RandomState | np.random.Generator)
-            and np.isnan(output[0]).any()
-        ):
+        if not isinstance(output[0], np.random.Generator) and np.isnan(output[0]).any():
             print("*** NaN detected ***")  # noqa: T201
             debugprint(node)
             print(f"Inputs : {[input[0] for input in fn.inputs]}")  # noqa: T201

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/compile/nanguardmode.py

@@ -34,7 +34,7 @@ def _is_numeric_value(arr, var):
 
     if isinstance(arr, _cdata_type):
         return False
-    elif isinstance(arr, np.random.mtrand.RandomState | np.random.Generator):
+    elif isinstance(arr, np.random.Generator):
         return False
     elif var is not None and isinstance(var.type, RandomType):
         return False

pytensor-2.31.0/pytensor/d3viz/html/template.html

@@ -0,0 +1,101 @@
+<!DOCTYPE html>
+<html>
+<head>
+	<link rel="stylesheet" href="%% CSS_DIR %%/d3viz.css"/>
+	<link rel="stylesheet" href="%% CSS_DIR %%/d3-context-menu.css"/>
+	<script type="text/javascript" src="%% JS_DIR %%/d3viz.js"></script>
+	<script type="text/javascript" src="%% JS_DIR %%/d3.v3.min.js"></script>
+	<script type="text/javascript" src="%% JS_DIR %%/dagre-d3.min.js"></script>
+	<script type="text/javascript" src="%% JS_DIR %%/graphlib-dot.min.js"></script>
+	<script type="text/javascript" src="%% JS_DIR %%/d3-context-menu.js"></script>
+</head>
+<body>
+	<div id='menu' class='menuBar'>
+	    <input name="resetNodes" 
+	           type="button" 
+	           value="Reset nodes"
+	           onclick="resetNodes()"/>
+	    <input name="releaseNodes" 
+	           type="button" 
+	           value="Release nodes"
+	           onclick="releaseNodes()"/>
+	</div>
+
+	<script type="text/javascript">
+		// Backend graph in DOT format
+		var dotGraph = graphlibDot.read(%% DOT_GRAPH %%);
+		// Frontend graph for visualization
+		var graph = {};
+		
+		var forceLayout;
+		var isProfiled = false; // true if profiling information available
+		var useProfileColors = false;
+		var fixOnInit = true; // fix nodes on initialization
+		var maxProfilePer = 0;
+		var profileColors = ["#fff5f0","#fee0d2","#fcbba1","#fc9272","#fb6a4a","#ef3b2c","#cb181d","#a50f15"];
+		var pad = 10;
+		var isEditNode = false; // true if node is edited
+		var menuItems = [
+			{
+				title: 'Edit',
+				action: function(elm, d, i) { editNode(elm, d); }
+			},
+			{
+				title: 'Release',
+				action: function(elm, d, i) { releaseNode(d); }
+			}
+		];
+
+		// Create main panel
+		d3.select('body').select('svg').remove();
+		var svg = d3.select('body').append('svg')
+			.attr('width', '100%')
+			.attr('height', '95%');
+		var pane = svg.append('g');
+		
+		// Zoom behaviour
+		var zoom = d3.behavior.zoom()
+			.scaleExtent([0.2, 8])
+			.on('zoom', function(d) {
+				var trans = d3.event.translate;
+				trans[0] += 300;
+				trans[1] += 100;
+				pane.attr('transform', 'translate(' + trans + ') scale(' + d3.event.scale + ')');
+			});
+		svg.call(zoom);
+		zoom.event(svg);
+		svg.on("dblclick.zoom", null);
+		
+		// Edges
+		var edgeDiv = d3.select('body').append('div')
+			.attr('class', 'edgeTooltip')
+			.style('opacity', 0.0);
+		
+		// Div for node details	
+		var nodeInfo = d3.select('body').append('div')
+			.attr('class', 'nodeInfo')
+			.style('opacity', 0.0);
+		
+		// Definition head of edges
+		var markerData = [
+			{'id': 'n', 'color': 'black'},
+			{'id': 'r', 'color': 'red'},
+			{'id': 'b', 'color': 'dodgerblue'}];
+		svg.append("defs").selectAll('marker').data(markerData).enter().append("marker")
+		    .attr("id", function(d) { return 'edgeArrow_' + d.id;})
+		    .attr("markerWidth", 4)
+		    .attr("markerHeight", 4)
+		    .attr("refX", 2)
+		    .attr("refY", 2)
+		    .attr("orient", "auto")
+		  	.append("path")
+		    .attr("d", "M0,0 L4,2 L0,4 Z")
+		    .attr('fill', function(d) { return d.color;});
+	
+		// Initialize graph
+		processDotGraph(dotGraph);		
+		graph = frontEndGraph(dotGraph);
+		drawGraph();
+	</script>
+</body>
+</html>

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/graph/op.py

@@ -502,7 +502,7 @@ class Op(MetaObject):
         self,
         node: Apply,
         storage_map: StorageMapType,
-        compute_map: ComputeMapType,
+        compute_map: ComputeMapType | None,
         no_recycling: list[Variable],
         debug: bool = False,
     ) -> ThunkType:
@@ -513,25 +513,38 @@ class Op(MetaObject):
         """
         node_input_storage = [storage_map[r] for r in node.inputs]
         node_output_storage = [storage_map[r] for r in node.outputs]
-        node_compute_map = [compute_map[r] for r in node.outputs]
 
         if debug and hasattr(self, "debug_perform"):
             p = node.op.debug_perform
         else:
             p = node.op.perform
 
-        @is_thunk_type
-        def rval(
-            p=p,
-            i=node_input_storage,
-            o=node_output_storage,
-            n=node,
-            cm=node_compute_map,
-        ):
-            r = p(n, [x[0] for x in i], o)
-            for entry in cm:
-                entry[0] = True
-            return r
+        if compute_map is None:
+
+            @is_thunk_type
+            def rval(
+                p=p,
+                i=node_input_storage,
+                o=node_output_storage,
+                n=node,
+            ):
+                return p(n, [x[0] for x in i], o)
+
+        else:
+            node_compute_map = [compute_map[r] for r in node.outputs]
+
+            @is_thunk_type
+            def rval(
+                p=p,
+                i=node_input_storage,
+                o=node_output_storage,
+                n=node,
+                cm=node_compute_map,
+            ):
+                r = p(n, [x[0] for x in i], o)
+                for entry in cm:
+                    entry[0] = True
+                return r
 
         rval.inputs = node_input_storage
         rval.outputs = node_output_storage

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/ifelse.py

@@ -305,8 +305,8 @@ class IfElse(_NoPythonOp):
                     if len(ls) > 0:
                         return ls
                     else:
-                        # strict=False because we are in a hot loop
-                        for out, t in zip(outputs, input_true_branch, strict=False):
+                        # zip strict not specified because we are in a hot loop
+                        for out, t in zip(outputs, input_true_branch):
                             compute_map[out][0] = 1
                             val = storage_map[t][0]
                             if self.as_view:
@@ -326,8 +326,8 @@ class IfElse(_NoPythonOp):
                     if len(ls) > 0:
                         return ls
                     else:
-                        # strict=False because we are in a hot loop
-                        for out, f in zip(outputs, inputs_false_branch, strict=False):
+                        # zip strict not specified because we are in a hot loop
+                        for out, f in zip(outputs, inputs_false_branch):
                             compute_map[out][0] = 1
                             # can't view both outputs unless destroyhandler
                             # improves

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/link/basic.py

@@ -539,14 +539,14 @@ class WrapLinker(Linker):
 
         def f():
             for inputs in input_lists[1:]:
-                # strict=False because we are in a hot loop
-                for input1, input2 in zip(inputs0, inputs, strict=False):
+                # zip strict not specified because we are in a hot loop
+                for input1, input2 in zip(inputs0, inputs):
                     input2.storage[0] = copy(input1.storage[0])
             for x in to_reset:
                 x[0] = None
             pre(self, [input.data for input in input_lists[0]], order, thunk_groups)
-            # strict=False because we are in a hot loop
-            for i, (thunks, node) in enumerate(zip(thunk_groups, order, strict=False)):
+            # zip strict not specified because we are in a hot loop
+            for i, (thunks, node) in enumerate(zip(thunk_groups, order)):
                 try:
                     wrapper(self.fgraph, i, node, *thunks)
                 except Exception:
@@ -668,8 +668,8 @@ class JITLinker(PerformLinker):
                 #  since the error may come from any of them?
                 raise_with_op(self.fgraph, output_nodes[0], thunk)
 
-            # strict=False because we are in a hot loop
-            for o_storage, o_val in zip(thunk_outputs, outputs, strict=False):
+            # zip strict not specified because we are in a hot loop
+            for o_storage, o_val in zip(thunk_outputs, outputs):
                 o_storage[0] = o_val
 
         thunk.inputs = thunk_inputs

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/link/c/basic.py

@@ -1988,27 +1988,23 @@ class DualLinker(Linker):
         )
 
         def f():
-            # strict=False because we are in a hot loop
-            for input1, input2 in zip(i1, i2, strict=False):
+            # zip strict not specified because we are in a hot loop
+            for input1, input2 in zip(i1, i2):
                 # Set the inputs to be the same in both branches.
                 # The copy is necessary in order for inplace ops not to
                 # interfere.
                 input2.storage[0] = copy(input1.storage[0])
-            for thunk1, thunk2, node1, node2 in zip(
-                thunks1, thunks2, order1, order2, strict=False
-            ):
-                for output, storage in zip(node1.outputs, thunk1.outputs, strict=False):
+            for thunk1, thunk2, node1, node2 in zip(thunks1, thunks2, order1, order2):
+                for output, storage in zip(node1.outputs, thunk1.outputs):
                     if output in no_recycling:
                         storage[0] = None
-                for output, storage in zip(node2.outputs, thunk2.outputs, strict=False):
+                for output, storage in zip(node2.outputs, thunk2.outputs):
                     if output in no_recycling:
                         storage[0] = None
                 try:
                     thunk1()
                     thunk2()
-                    for output1, output2 in zip(
-                        thunk1.outputs, thunk2.outputs, strict=False
-                    ):
+                    for output1, output2 in zip(thunk1.outputs, thunk2.outputs):
                         self.checker(output1, output2)
                 except Exception:
                     raise_with_op(fgraph, node1)

{pytensor-2.30.2 → pytensor-2.31.0}/pytensor/link/c/c_code/lazylinker_c.c

@@ -477,7 +477,7 @@ static PyObject *pycall(CLazyLinker *self, Py_ssize_t node_idx, int verbose) {
     double t0 = pytime(NULL);
     if (verbose)
       fprintf(stderr, "calling via Python (node %i)\n", (int)node_idx);
-    rval = PyObject_CallObject(thunk, NULL);
+    rval = PyObject_CallNoArgs(thunk);
     if (rval) {
       double t1 = pytime(NULL);
       double ti = PyFloat_AsDouble(PyList_GetItem(self->call_times, node_idx));
@@ -491,7 +491,7 @@ static PyObject *pycall(CLazyLinker *self, Py_ssize_t node_idx, int verbose) {
     if (verbose) {
       fprintf(stderr, "calling via Python (node %i)\n", (int)node_idx);
     }
-    rval = PyObject_CallObject(thunk, NULL);
+    rval = PyObject_CallNoArgs(thunk);
   }
   return rval;
 }
@@ -981,7 +981,7 @@ static PyTypeObject lazylinker_ext_CLazyLinkerType = {
 };
 
 static PyObject *get_version(PyObject *dummy, PyObject *args) {
-  PyObject *result = PyFloat_FromDouble(0.3);
+  PyObject *result = PyFloat_FromDouble(0.31);
   return result;
 }