jupyter#751 handling widget metadata without model_name

twosigma · Mar 15, 2018 · a301aa4 · a301aa4
1 parent b5e9997
commit a301aa4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nbconvert/preprocessors/execute.py b/nbconvert/preprocessors/execute.py
@@ -286,7 +286,7 @@ def start_new_kernel(startup_timeout=60, kernel_name='python', **kwargs):
                     'application/vnd.jupyter.widget-state+json': {
                         'state': {
                             model_id: _serialize_widget_state(state)
-                            for model_id, state in self.widget_state.items()
+                            for model_id, state in self.widget_state.items() if '_model_name' in state
                         },
                         'version_major': 2,
                         'version_minor': 0,
@@ -486,8 +486,8 @@ def _serialize_widget_state(state):
     TODO: Does not currently split binary buffers or remove default values.
     """
     return {
-        'model_name': state['_model_name'],
-        'model_module': state['_model_module'],
+        'model_name': state.get('_model_name'),
+        'model_module': state.get('_model_module'),
         'model_module_version': state.get('_model_module_version'),
         'state': state,
     }