Skip to content

Commit 97c9379

Browse files
Run pre-commit
1 parent 5023274 commit 97c9379

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pytensor/link/numba/dispatch/slinalg.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from pytensor.link.numba.dispatch import basic as numba_basic
1111
from pytensor.link.numba.dispatch.basic import numba_funcify
12-
from pytensor.tensor.slinalg import Cholesky, BlockDiagonal, SolveTriangular
12+
from pytensor.tensor.slinalg import BlockDiagonal, Cholesky, SolveTriangular
1313

1414

1515
_PTR = ctypes.POINTER
@@ -299,6 +299,7 @@ def solve_triangular(a, b):
299299

300300
return solve_triangular
301301

302+
302303
def _cholesky(a, lower=False, overwrite_a=False, check_finite=True):
303304
return linalg.cholesky(
304305
a, lower=lower, overwrite_a=overwrite_a, check_finite=check_finite
@@ -357,6 +358,7 @@ def nb_cholesky(a):
357358

358359
return nb_cholesky
359360

361+
360362
@numba_funcify.register(BlockDiagonal)
361363
def numba_funcify_BlockDiagonal(op, node, **kwargs):
362364
dtype = node.outputs[0].dtype

0 commit comments

Comments
 (0)