Skip to content

Commit 306df11

Browse files
pre-commit-ci[bot]jessegrabowski
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d31af6e commit 306df11

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

pytensor/tensor/basic.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -663,6 +663,7 @@ def c_code_cache_version(self):
663663

664664
tensor_from_scalar = TensorFromScalar()
665665

666+
666667
@_vectorize_node.register(TensorFromScalar)
667668
def vectorize_tensor_from_scalar(op, node, batch_x):
668669
return identity(batch_x).owner

pytensor/tensor/optimize.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -561,7 +561,10 @@ def L_op(self, inputs, outputs, output_grads):
561561
implicit_f = grad(inner_fx, inner_x)
562562

563563
df_dx, *df_dtheta_columns = jacobian(
564-
implicit_f, [inner_x, *inner_args], disconnected_inputs="ignore", vectorize=True
564+
implicit_f,
565+
[inner_x, *inner_args],
566+
disconnected_inputs="ignore",
567+
vectorize=True,
565568
)
566569
grad_wrt_args = implict_optimization_grads(
567570
df_dx=df_dx,

0 commit comments

Comments
 (0)