Skip to content

Commit 8dc2a26

Browse files
Resolve conflicts with main
1 parent b68b973 commit 8dc2a26

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

pytensor/link/numba/dispatch/basic.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
from pytensor.tensor.blas import BatchedDot
3838
from pytensor.tensor.math import Dot
3939
from pytensor.tensor.shape import Reshape, Shape, Shape_i, SpecifyShape
40-
from pytensor.tensor.slinalg import Solve
40+
from pytensor.tensor.slinalg import Cholesky, Solve
4141
from pytensor.tensor.subtensor import (
4242
AdvancedIncSubtensor,
4343
AdvancedIncSubtensor1,
@@ -808,6 +808,7 @@ def softplus(x):
808808

809809
return softplus
810810

811+
811812
@numba_funcify.register(Cholesky)
812813
def numba_funcify_Cholesky(op, node, **kwargs):
813814
lower = op.lower

pytensor/tensor/slinalg.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,11 +50,9 @@ class Cholesky(Op):
5050
__props__ = ("lower", "overwrite_a", "on_error")
5151
gufunc_signature = "(m,m)->(m,m)"
5252

53-
def __init__(self, *, lower=True, on_error="raise", overwrite_a=False):
54-
self.lower = lower
55-
self.overwrite_a = overwrite_a
56-
57-
def __init__(self, *, lower=True, check_finite=True, on_error="raise", overwrite_a=False):
53+
def __init__(
54+
self, *, lower=True, check_finite=True, on_error="raise", overwrite_a=False
55+
):
5856
self.lower = lower
5957
self.destructive = False
6058
self.check_finite = check_finite
@@ -154,11 +152,12 @@ def try_inplace_inputs(self, candidate_inputs: list[int]) -> "Op":
154152
lower=self.lower, overwrite_a=True, on_error=self.on_error
155153
)
156154

155+
157156
def cholesky(x, lower=True, on_error="raise", check_finite=False):
158157
return Blockwise(
159158
Cholesky(lower=lower, on_error=on_error, check_finite=check_finite)
160159
)(x)
161-
160+
162161

163162
class SolveBase(Op):
164163
"""Base class for `scipy.linalg` matrix equation solvers."""

0 commit comments

Comments
 (0)