Skip to content

Commit 497721e

Browse files
Remove more useful type information for mypy
1 parent 4bd259c commit 497721e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pytensor/tensor/slinalg.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
import pytensor
1313
import pytensor.tensor as pt
14-
from pytensor import Variable
1514
from pytensor.gradient import DisconnectedType
1615
from pytensor.graph.basic import Apply
1716
from pytensor.graph.op import Op
@@ -1715,7 +1714,7 @@ def perform(self, node, inputs, outputs_storage):
17151714
fn = scipy_linalg.get_blas_funcs("gbmv", dtype=A.dtype)
17161715
outputs_storage[0][0] = fn(m=m, n=n, kl=kl, ku=ku, alpha=1, a=A_banded, x=x)
17171716

1718-
def L_op(self, inputs, outputs, output_grads) -> list[Variable]:
1717+
def L_op(self, inputs, outputs, output_grads):
17191718
# This is exactly the same as the usual gradient of a matrix-vector product, except that the banded structure
17201719
# is exploited.
17211720
A, x = inputs

0 commit comments

Comments
 (0)