compatibility fix for isort

koxudaxi · Nov 8, 2020 · 1f56ce4 · 1f56ce4
1 parent bddf61a
commit 1f56ce4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datamodel_code_generator/format.py b/datamodel_code_generator/format.py
@@ -4,7 +4,7 @@
 
 import black
 import toml
-from isort import SortImports
+import isort
 
 
 class PythonVersion(Enum):
@@ -49,4 +49,4 @@ def apply_black(code: str, python_version: PythonVersion) -> str:
 
 
 def apply_isort(code: str) -> str:
-    return SortImports(file_contents=code).output
+    return isort.code(code)