Run tests in docker container, correct jax dependency

FLAIROx / JaxMARL

Multi-Agent Reinforcement Learning with JAX

Apache License 2.0

393 stars 68 forks source link

Run tests in docker container, correct jax dependency #75

Closed amacrutherford closed 5 months ago

amacrutherford commented 5 months ago

Change push and pr tests to be run in our dockerfile using make workflow-test. Update explicit jax dependency as this is causing jax to be reinstalled during make build which reverts jax back to a CPU version