Skip to content

Commit 2fd7a75

Browse files
committed
numba while condition must be tensor
1 parent ca14e24 commit 2fd7a75

File tree

1 file changed

+2
-2
lines changed
  • pytensor/link/numba/dispatch

1 file changed

+2
-2
lines changed

pytensor/link/numba/dispatch/scan.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -351,8 +351,8 @@ def scan({", ".join(outer_in_names)}):
351351
{indent(input_storage_block, " " * 4)}
352352
353353
i = 0
354-
cond = False
355-
while i < n_steps and not cond:
354+
cond = np.array(False)
355+
while i < n_steps and not cond.item():
356356
{inner_outputs} = scan_inner_func({inner_in_args})
357357
{indent(inner_out_post_processing_block, " " * 8)}
358358
{indent(inner_out_to_outer_out_stmts, " " * 8)}

0 commit comments

Comments
 (0)