Skip to content

Commit c88b70b

Browse files
committed
Change py_only database placement
1 parent 6d3c756 commit c88b70b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytensor/compile/mode.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -251,13 +251,14 @@ def apply(self, fgraph):
251251
# especially constant merge
252252
optdb.register("merge2", MergeOptimizer(), "fast_run", "merge", position=49)
253253

254+
optdb.register("py_only", EquilibriumDB(), "fast_compile", position=49.1)
255+
254256
optdb.register(
255257
"add_destroy_handler", AddDestroyHandler(), "fast_run", "inplace", position=49.5
256258
)
257259

258260
# final pass just to make sure
259261
optdb.register("merge3", MergeOptimizer(), "fast_run", "merge", position=100)
260-
optdb.register("py_only", EquilibriumDB(), "fast_compile", position=100)
261262

262263
_tags: Union[Tuple[str, str], Tuple]
263264

0 commit comments

Comments
 (0)