Skip to content

Commit 70eec3c

Browse files
committed
Respect predefined modes in get_default_mode
Also make linker and optimizer non-mutable config as the mode is cached after using them for the first time.
1 parent 15a887a commit 70eec3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/compile/mode.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,13 +535,13 @@ def get_mode(orig_string):
535535
ret = Mode(linker=config.linker, optimizer=config.optimizer)
536536

537537
if orig_string is None:
538-
# Build and cache the ~~default~~ first requested mode
539538
if config.optimizer_excluding:
540539
ret = ret.excluding(*config.optimizer_excluding.split(":"))
541540
if config.optimizer_including:
542541
ret = ret.including(*config.optimizer_including.split(":"))
543542
if config.optimizer_requiring:
544543
ret = ret.requiring(*config.optimizer_requiring.split(":"))
544+
# Override the cache with the new class mode
545545
instantiated_default_mode = ret
546546

547547
return ret

0 commit comments

Comments
 (0)