Skip to content

Commit 5ad1181

Browse files
committed
Fix memory leak in TensorFromScalar
1 parent c9159b2 commit 5ad1181

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pytensor/tensor/basic.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -593,16 +593,15 @@ def c_code(self, node, name, inputs, outputs, sub):
593593
return (
594594
"""
595595
%(z)s = (PyArrayObject*)PyArray_FromScalar(py_%(x)s, NULL);
596-
if(py_%(z)s == NULL){
596+
if(%(z)s == NULL){
597597
%(fail)s;
598598
}
599-
Py_XINCREF(%(z)s);
600599
"""
601600
% locals()
602601
)
603602

604603
def c_code_cache_version(self):
605-
return (1,)
604+
return (2,)
606605

607606

608607
tensor_from_scalar = TensorFromScalar()

0 commit comments

Comments
 (0)