Numba scan: fix typo

ricardoV94 · twiecki · commit e5bdedb860dc · 2023-05-10T10:09:01.000+02:00
diff --git a/pytensor/link/numba/dispatch/scan.py b/pytensor/link/numba/dispatch/scan.py
@@ -367,8 +367,6 @@ def scan({", ".join(outer_in_names)}):
     }
     global_env["np"] = np
 
-    scalar_op_fn = compile_function_src(
-        scan_op_src, "scan", {**globals(), **global_env}
-    )
+    scan_op_fn = compile_function_src(scan_op_src, "scan", {**globals(), **global_env})
 
-    return numba_basic.numba_njit(scalar_op_fn)
+    return numba_basic.numba_njit(scan_op_fn)

Original file line number	Diff line number	Diff line change
`@@ -367,8 +367,6 @@ def scan({", ".join(outer_in_names)}):`
`367`	`367`	`}`
`368`	`368`	`global_env["np"] = np`
`369`	`369`
`370`		`- scalar_op_fn = compile_function_src(`
`371`		`- scan_op_src, "scan", {globals(), global_env}`
`372`		`- )`
	`370`	`+ scan_op_fn = compile_function_src(scan_op_src, "scan", {globals(), global_env})`
`373`	`371`
`374`		`- return numba_basic.numba_njit(scalar_op_fn)`
	`372`	`+ return numba_basic.numba_njit(scan_op_fn)`