Add back _remove_method

glue-viz · Jan 25, 2019 · e675dd0 · e675dd0
1 parent ed6562f
commit e675dd0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/glue/viewers/image/frb_artist.py b/glue/viewers/image/frb_artist.py
@@ -53,4 +53,12 @@ def imshow(axes, X, aspect=None, vmin=None, vmax=None, **kwargs):
     if vmin is not None or vmax is not None:
         im.set_clim(vmin, vmax)
 
+    axes.images.append(im)
+
+    def remove(h):
+        axes.images.remove(h)
+
+    # The following is needed for Matplotlib's .remove() method to work
+    im._remove_method = remove
+
     return im