From f3dc85c3e84a003f51f61360f89b652e0f21b1e4 Mon Sep 17 00:00:00 2001 From: Jacob Silterra Date: Sun, 28 Jul 2024 17:31:41 -0400 Subject: [PATCH] Bugfix for InferencerPlotter in 1D `outvar` is a dict so need to select the appropriate key. Signed-off-by: Jacob Silterra --- modulus/sym/utils/io/plotter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modulus/sym/utils/io/plotter.py b/modulus/sym/utils/io/plotter.py index bbc41e69..2ca431e1 100644 --- a/modulus/sym/utils/io/plotter.py +++ b/modulus/sym/utils/io/plotter.py @@ -139,7 +139,7 @@ def __call__(self, invar, outvar): for k in outvar: f = plt.figure(figsize=(5, 4), dpi=100) if ndim == 1: - plt.plot(invar[dims[0]][:, 0], outvar[:, 0]) + plt.plot(invar[dims[0]][:, 0], outvar[k][:, 0]) plt.xlabel(dims[0]) elif ndim == 2: plt.imshow(outvar[k].T, origin="lower", extent=extent)