tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

refactor: convert jax verification script to pytest and update docs

Browse source
This commit is contained in:
Tibo De Peuter 2026-03-21 16:53:07 +01:00
parent 741aafab07
commit cb5a6fbd87
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
3 changed files with 43 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

6
docs/DEVELOPMENT.md
View file

@ -53,9 +53,9 @@ If you prefer not to use Docker:
## Hardware Acceleration (JAX)
Verify your setup by running:
Verify your setup by running the JAX initialization test:
```bash
python -c "import jax; print(jax.devices())"
uv run pytest tests/test_jax_init.py
```
In the devcontainer, this will report a `GpuDevice` if a GPU is detected and the `cuda` extra was installed.
In the devcontainer, this will succeed on both CPU and GPU. A `GpuDevice` is expected if a GPU is detected and the `cuda` extra was installed.