TypeError: expected str, bytes or os.PathLike object, not NoneType

[CharonRob]

I installed a series of packages such as jax according to the readme, but when I ran bash run_learner.sh, an error occurred:

import jax 

TypeError: expected str, bytes or os.PathLike object, not NoneType

[violasox]

I ran into this same issue, and it seems like this is an issue with jax==0.4.35. I can import jax successfully with other versions (e.g. 0.4.36, 0.6.2). Unfortunately, the codebase relies on jax.experimental.maps, which seems to be removed as of 0.4.36. So fixing this might require modifications to the codebase.

Related issue: jax-ml/jax#30109