Skip to content

Commit fee1a43

Browse files
Jammy2211Jammy2211
authored andcommitted
Fix subplot_of_mapper crash on interferometer data_subtracted
Panel 0 of `subplot_of_mapper` and `subplot_mappings` passed `inversion.data_subtracted_dict[mapper]` directly to `plot_array`, which expects a 2D array. For interferometer fits the entry is `Visibilities` (1D), so `plot_array`'s `array.shape[:2]` unpack failed. Mirror the existing `Visibilities` check in panels 1-3: when the dict entry is `Visibilities`, transform to a 2D dirty image via `inversion.transformer.image_from(...)` (same mechanism as `FitInterferometer.dirty_residual_map`). Imaging path unchanged.
1 parent ae43ae1 commit fee1a43

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

autoarray/inversion/plot/inversion_plots.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,13 @@ def subplot_of_mapper(
6060

6161
# panel 0: data subtracted
6262
try:
63+
array = inversion.data_subtracted_dict[mapper]
64+
from autoarray.structures.visibilities import Visibilities
65+
66+
if isinstance(array, Visibilities):
67+
array = inversion.transformer.image_from(visibilities=array)
6368
plot_array(
64-
inversion.data_subtracted_dict[mapper],
69+
array,
6570
ax=axes[0],
6671
title=_pf("Data Subtracted"),
6772
colormap=colormap,
@@ -286,8 +291,13 @@ def subplot_mappings(
286291

287292
# panel 0: data subtracted
288293
try:
294+
array = inversion.data_subtracted_dict[mapper]
295+
from autoarray.structures.visibilities import Visibilities
296+
297+
if isinstance(array, Visibilities):
298+
array = inversion.transformer.image_from(visibilities=array)
289299
plot_array(
290-
inversion.data_subtracted_dict[mapper],
300+
array,
291301
ax=axes[0],
292302
title="Data Subtracted",
293303
colormap=colormap,

0 commit comments

Comments
 (0)