handle ValueError (#20)

* handle ValueError * Fix MutableMapping import on Python 3.10
stitchfix · Jul 28, 2022 · fc71034 · fc71034
1 parent 6482a7b
commit fc71034
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/nodebook/ipython/nodebookext.py b/nodebook/ipython/nodebookext.py
@@ -67,7 +67,11 @@ def execute_cell(line, cell):
     ipython magic for executing nodebook cell, expects cell id and parent id inline, followed by code
     """
     assert NODEBOOK_STATE['nodebook'] is not None, "Nodebook not initialized, please use %nodebook {nodebook_name}"
-    cell_id, parent_id = line.lstrip().split(' ')
+    try:
+        cell_id, parent_id = line.lstrip().split(' ')
+    except ValueError:
+        cell_id = line.lstrip()
+        parent_id = None
 
     # make sure cell exists and is in the right position
     NODEBOOK_STATE['nodebook'].insert_node_after(cell_id, parent_id)

diff --git a/nodebook/pickledict.py b/nodebook/pickledict.py
@@ -26,7 +26,7 @@
     from UserDict import DictMixin
 except ImportError:
     # see https://github.com/flask-restful/flask-restful/pull/231/files
-    from collections import MutableMapping as DictMixin
+    from collections.abc import MutableMapping as DictMixin
 
 PANDAS_CODE = 1
 CLOUDPICKLE_CODE = 2