Incorrect slice

Author: jessegrabowski

pytensor/tensor/rewriting/linalg.py

@@ -912,7 +912,7 @@ def rewrite_cholesky_diag_to_sqrt_diag(fgraph, node):
     [input] = node.inputs
 
     # Check if input is a (1, 1) matrix
-    if all(input.type.broadcastable[:-2]):
+    if all(input.type.broadcastable[-2:]):
         return [pt.sqrt(input)]
 
     # Check for use of pt.diag first