fix: address comments

Signed-off-by: yihong0618 <[email protected]>

Author: yihong0618

Lib/timeit.py

@@ -133,7 +133,7 @@ def __init__(self, stmt="pass", setup="pass", timer=default_timer,
         exec(code, global_ns, local_ns)
         self.inner = local_ns["inner"]
 
-    def print_exc(self, file=None):
+    def print_exc(self, file=None, **kwargs):
         """Helper to print a traceback from the timed code.
 
         Typical use:
@@ -158,8 +158,10 @@ def print_exc(self, file=None):
                                                dummy_src_name)
         # else the source is already stored somewhere else
 
-        traceback.print_exception(sys.exception(), file=file,
-                                 colorize=_colorize.can_colorize(file=file))
+        if 'colorize' not in kwargs:
+            kwargs['colorize'] = _colorize.can_colorize(file=file)
+
+        traceback.print_exc(file=file, **kwargs)
 
     def timeit(self, number=default_number):
         """Time 'number' executions of the main statement.

Lib/traceback.py

@@ -206,9 +206,9 @@ def _safe_string(value, what, func=str):
 
 # --
 
-def print_exc(limit=None, file=None, chain=True):
+def print_exc(limit=None, file=None, chain=True, **kwargs):
     """Shorthand for 'print_exception(sys.exception(), limit=limit, file=file, chain=chain)'."""
-    print_exception(sys.exception(), limit=limit, file=file, chain=chain)
+    print_exception(sys.exception(), limit=limit, file=file, chain=chain, **kwargs)
 
 def format_exc(limit=None, chain=True):
     """Like print_exc() but return a string."""