AttributeError: module 'jax.api' has no attribute '_jit_is_disabled'

[mmcaulif]

Hi, unsure if this is a due to coax or jax but I get this error when running the pendulum ppo example, dqn runs fine however.

A similar error I found online recommended changing the version of jaxlib so I went to using the jaxlib version set out in the coax getting started guide but seemed to have no affect
jax version = 0.2.13
jaxlib version = 0.1.65 + cuda111
coax version = 0.1.6

[KristianHolsheimer]

Thanks for your interest in coax!

I'm traveling at the moment, with no access to a computer and limited internet access. But from memory, I believe that I removed any access to internal attrs in the latest version (main branch).

Could you try installing coax directly from the main branch?

pip install git+https://github.com/coax-dev/coax.git@main

Please let me know if that fixes it.

[mmcaulif]

Thanks for your interest in coax!

I'm traveling at the moment, with no access to a computer and limited internet access. But from memory, I believe that I removed any access to internal attrs in the latest version (main branch).

Could you try installing coax directly from the main branch?

pip install git+https://github.com/coax-dev/coax.git@main

Please let me know if that fixes it.

That did the job!

I think Coax is a great educational tool and perfectly bridges the gap from the theory and the practical coding of RL :)

[KristianHolsheimer]

Okay, I'll bump the version to upgrade the package on pypi.

Thanks for the kind words, that made my day!