From ac3709a7a826a6c7ec123e9ac5d95189842c18ef Mon Sep 17 00:00:00 2001 From: Maple-YZ Date: Fri, 10 Apr 2026 01:24:40 +0800 Subject: [PATCH] =?UTF-8?q?=E9=81=BF=E5=85=8D=E7=BB=98=E5=88=B6=E7=83=AD?= =?UTF-8?q?=E5=8A=9B=E5=9B=BE=E6=97=B6=E8=87=AA=E5=8A=A8=E5=90=AF=E7=94=A8?= =?UTF-8?q?=E6=A8=A1=E7=B3=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/episode_artifacts.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/utils/episode_artifacts.py b/utils/episode_artifacts.py index 2a0e263..5503cba 100644 --- a/utils/episode_artifacts.py +++ b/utils/episode_artifacts.py @@ -110,7 +110,14 @@ def _plot_episode_heatmap(path: str, edge_rows: Sequence[Dict], edge_ids: Sequen ] for ax, (grid, cmap, title) in zip(axes, plots): - image = ax.imshow(np.ma.masked_invalid(grid), aspect="auto", origin="lower", cmap=cmap) + image = ax.imshow( + np.ma.masked_invalid(grid), + aspect="auto", + origin="lower", + cmap=cmap, + interpolation="nearest", + resample=False, + ) ax.set_title(f"{title_prefix} {title}") ax.set_xlabel("Decision Step") ax.set_ylabel("Controlled Edge")