search

coax-dev / coax

Modular framework for Reinforcement Learning in python
https://coax.readthedocs.io
MIT License
168 stars 17 forks source link

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

Closed mmcaulif closed 3 years ago

mmcaulif commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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

Thanks for the kind words, that made my day!