diff --git a/pyproject.toml b/pyproject.toml index bbcab6c..a379856 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "thants" -version = "0.3.2" +version = "0.3.3" description = "Ants! On your GPU! With JAX!" authors = ["zombie-einstein "] license = "MIT" diff --git a/src/thants/viewer.py b/src/thants/viewer.py index c36e1a4..02aca58 100644 --- a/src/thants/viewer.py +++ b/src/thants/viewer.py @@ -204,7 +204,7 @@ def _draw( ax.imshow(terrain) ax.imshow(nests) - food_img = ax.imshow(jnp.clip(state.food, 0.0, 1.0), alpha=state.food) + food_img = ax.imshow(food, alpha=jnp.clip(state.food, 0.0, 1.0)) ants_img = ax.imshow(ants) return colors, ants_img, food_img