Skip to content

Commit 60b3e89

Browse files
committed
[bug] fix analysis jit bug
1 parent 20819c4 commit 60b3e89

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

brainpy/_src/analysis/lowdim/lowdim_analyzer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -436,7 +436,7 @@ def F_fy(self):
436436
f = partial(f, **(self.pars_update + self.fixed_vars))
437437
f = utils.f_without_jaxarray_return(f)
438438
f = utils.remove_return_shape(f)
439-
self.analyzed_results[C.F_fy] = bm.jit(f, device=self.jit_device)
439+
self.analyzed_results[C.F_fy] = jax.jit(f, device=self.jit_device)
440440
return self.analyzed_results[C.F_fy]
441441

442442
@property

0 commit comments

Comments
 (0)