diff options
Diffstat (limited to 'src/python/m5/util')
| -rw-r--r-- | src/python/m5/util/code_formatter.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/python/m5/util/code_formatter.py b/src/python/m5/util/code_formatter.py index c845c9021..21bbcd7a8 100644 --- a/src/python/m5/util/code_formatter.py +++ b/src/python/m5/util/code_formatter.py @@ -26,7 +26,11 @@ from __future__ import print_function -import __builtin__ +try: + import builtins +except ImportError: + # Python 2 fallback + import __builtin__ as builtins import inspect import os import re @@ -66,8 +70,8 @@ class lookup(object): if self.formatter.globals and item in self.frame.f_globals: return self.frame.f_globals[item] - if item in __builtin__.__dict__: - return __builtin__.__dict__[item] + if item in builtins.__dict__: + return builtins.__dict__[item] try: item = int(item) |