diff options
Diffstat (limited to 'clang/tools/clang-format/clang-format-diff.py')
-rwxr-xr-x | clang/tools/clang-format/clang-format-diff.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/clang/tools/clang-format/clang-format-diff.py b/clang/tools/clang-format/clang-format-diff.py index 54347ce7593..d6d35104161 100755 --- a/clang/tools/clang-format/clang-format-diff.py +++ b/clang/tools/clang-format/clang-format-diff.py @@ -28,10 +28,11 @@ import difflib import re import subprocess import sys -try: - from StringIO import StringIO -except ImportError: - from io import StringIO + +if sys.version_info.major >= 3: + from io import StringIO +else: + from io import BytesIO as StringIO def main(): |