Skip to content

Commit 30a71c4

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f20abb9 commit 30a71c4

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
@@ -560,7 +560,10 @@ def L_op(self, inputs, outputs, output_grads):
560560
implicit_f = grad(inner_fx, inner_x)
561561

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

0 commit comments

Comments
 (0)