[BUG] Flatland jax docker image doesn't work

[KaleabTessera]

Describe the bug

The current flatland docker image doesn't work.

Error: module 'jaxlib.xla_extension' has no attribute '__path__' due to version of cloudpickle that is required by gym 0.14 (flatland requires this version of gym).

To Reproduce

Steps to reproduce the behavior:

1. make build version=flatland-jax && make run example=examples/jax/flatland/feedforward/decentralised/run_mappo.py

Expected behavior

For it to run.

Context (Environment)

• Mava (and other package versions): result of pip freeze > requirements.txt
• OS

Additional context

Possible Solution