From b933cfdcb0b1e02e62c10d5404ea8085afbcf0b6 Mon Sep 17 00:00:00 2001
From: Min RK <benjaminrk@gmail.com>
Date: Wed, 14 Jan 2015 15:56:39 -0800
Subject: [PATCH 1/2] fix find_output_cell for IPython >= 3.0

output mime-type keys moved from top-level to a `data` attribute.
---
 lib/matplotlib/backends/web_backend/nbagg_mpl.js | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/lib/matplotlib/backends/web_backend/nbagg_mpl.js b/lib/matplotlib/backends/web_backend/nbagg_mpl.js
index 609f4e3a0b57..c22d385e65f6 100644
--- a/lib/matplotlib/backends/web_backend/nbagg_mpl.js
+++ b/lib/matplotlib/backends/web_backend/nbagg_mpl.js
@@ -160,12 +160,15 @@ mpl.find_output_cell = function(html_output) {
     var ncells = cells.length;
     for (var i=0; i<ncells; i++) {
         var cell = cells[i];
-        if (cell.cell_type == 'code'){
+        if (cell.cell_type === 'code'){
             for (var j=0; j<cell.output_area.outputs.length; j++) {
                 var data = cell.output_area.outputs[j];
-                if (cell.output_area.outputs[j]['text/html'] == html_output) {
-                    var output = cell.output_area.outputs[j];
-                    return [cell, output, j];
+                if (data.data) {
+                    // IPython >= 3 moved mimebundle to data attribute of output
+                    data = data.data;
+                }
+                if (data['text/html'] == html_output) {
+                    return [cell, data, j];
                 }
             }
         }

From 654afd27c27cc536ea22f549fe0db208a3e7d6b4 Mon Sep 17 00:00:00 2001
From: Min RK <benjaminrk@gmail.com>
Date: Wed, 14 Jan 2015 15:56:48 -0800
Subject: [PATCH 2/2] log failure to find cell/output

---
 lib/matplotlib/backends/web_backend/nbagg_mpl.js | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lib/matplotlib/backends/web_backend/nbagg_mpl.js b/lib/matplotlib/backends/web_backend/nbagg_mpl.js
index c22d385e65f6..fa333ddc974f 100644
--- a/lib/matplotlib/backends/web_backend/nbagg_mpl.js
+++ b/lib/matplotlib/backends/web_backend/nbagg_mpl.js
@@ -44,6 +44,10 @@ mpl.mpl_figure_comm = function(comm, msg) {
 
     fig.parent_element = element.get(0);
     fig.cell_info = mpl.find_output_cell("<div id='" + id + "'></div>");
+    if (!fig.cell_info) {
+        console.error("Failed to find cell for figure", id, fig);
+        return;
+    }
 
     var output_index = fig.cell_info[2]
     var cell = fig.cell_info[0];