From 3ff92638192d1b123c61bd28e478d1cbf065407a Mon Sep 17 00:00:00 2001 From: zombie-einstein <13398815+zombie-einstein@users.noreply.github.com> Date: Thu, 13 Nov 2025 11:24:57 +0000 Subject: [PATCH] Viewer bug fix --- pyproject.toml | 2 +- src/thants/viewer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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