Skip to content

Commit 1e96fd5

Browse files
Jammy2211Jammy2211
authored andcommitted
fix JAX env error
1 parent 13abefd commit 1e96fd5

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

autofit/jax_wrapper.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,7 @@
2222
if xla_env is None:
2323
xla_env_set = False
2424
elif isinstance(xla_env, str):
25-
xla_env_set = not "--xla_disable_hlo_passes=constant_folding" in xla_env
26-
25+
xla_env_set = "--xla_disable_hlo_passes=constant_folding" in xla_env
2726

2827
if not xla_env_set:
2928
logger.info(

0 commit comments

Comments
 (0)