Update code to work with isort5

essm/_generator.py

@@ -25,8 +25,8 @@
 import re
 from collections import defaultdict
 
+import isort
 import pkg_resources
-from isort import SortImports
 from sympy import Eq, Function, latex, preorder_traversal
 from sympy.physics.units import Quantity
 from yapf.yapflib.yapf_api import FormatCode
@@ -92,7 +92,7 @@ class {name}(Variable):
 
 def _lint_content(content):
     """Automatically lint the generated code."""
-    content = SortImports(file_contents=content).output
+    content = isort.code(content)
     content = FormatCode(content, style_config=STYLE_YAPF)[0]
     return content
 

setup.py

@@ -30,7 +30,7 @@
 tests_require = [
     'check-manifest>=0.25',
     'coverage>=4.0',
-    'isort>=4.2.2',
+    'isort>=5.0.0',
     'pydocstyle>=1.0.0,<4.0.0',
     'pytest-cache>=1.0',
     'pytest-cov>=1.8.0',