Merge pull request #20 from fchapoton/patch-1

do not use six
vbraun · Nov 13, 2020 · cb591d4 · cb591d4
2 parents 9bf36a0 + b2c1686
commit cb591d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -8,7 +8,6 @@
     author='Volker Braun',
     author_email='[email protected]',
     install_requires=[
-        'six',
         'ipython>=4',
         'nbconvert>=4',
         'notebook',

diff --git a/test/test_sagenb_reader.py b/test/test_sagenb_reader.py
@@ -1,17 +1,16 @@
 
 import os
 import unittest
-import six
 from sagenb_export.sagenb_reader import (
     NotebookSageNB,
     TextCell, ComputeCell,
 )
 
 
-if six.PY2:
-    string_type = unicode
-else:
-    string_type = str
+try:
+    string_type = unicode  # py2
+except NameError:
+    string_type = str  # py3
 
 
 DOT_SAGE = os.path.join(os.path.dirname(__file__), 'dot_sage')