diff --git a/python/tvm/runtime/script_printer.py b/python/tvm/runtime/script_printer.py index 2ed2b8ddd4bc..37e70b6f94f4 100644 --- a/python/tvm/runtime/script_printer.py +++ b/python/tvm/runtime/script_printer.py @@ -199,7 +199,7 @@ def script( def show( self, style: Optional[str] = None, - black_format: bool = True, + black_format: bool = False, *, name: Optional[str] = None, show_meta: bool = False, @@ -227,7 +227,7 @@ def show( Pygmentize printing style, auto-detected if None. See `tvm.script.highlight.cprint` for more details. black_format: bool - If true (default), use the formatter Black to format the TVMScript + If true, use the formatter Black to format the TVMScript name : Optional[str] = None The name of the object show_meta : bool = False diff --git a/python/tvm/script/highlight.py b/python/tvm/script/highlight.py index 80a1f9972248..be0de5a6bf2b 100644 --- a/python/tvm/script/highlight.py +++ b/python/tvm/script/highlight.py @@ -26,7 +26,7 @@ def cprint( printable: Union[Any, str], style: Optional[str] = None, - black_format: bool = True, + black_format: bool = False, ) -> None: """Print TVMScript string with Pygments highlight and Black auto-formatting. @@ -42,7 +42,7 @@ def cprint( black_format: bool - If true (default), use the formatter Black to format the TVMScript + If true, use the formatter Black to format the TVMScript Notes ----- diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py index e82cd251d33c..85d1f19bba00 100644 --- a/python/tvm/tir/schedule/schedule.py +++ b/python/tvm/tir/schedule/schedule.py @@ -268,7 +268,7 @@ def fork_seed(self) -> int: """ return _ffi_api.ScheduleForkSeed(self) # type: ignore # pylint: disable=no-member - def show(self, style: Optional[str] = None, black_format: bool = True) -> None: + def show(self, style: Optional[str] = None, black_format: bool = False) -> None: """A sugar for print highlighted TVM script. Parameters @@ -280,7 +280,7 @@ def show(self, style: Optional[str] = None, black_format: bool = True) -> None: black_format: bool - If true (default), use the formatter Black to format the TVMScript + If true, use the formatter Black to format the TVMScript """ mod = self.mod if mod is not None: diff --git a/python/tvm/tir/schedule/trace.py b/python/tvm/tir/schedule/trace.py index e40a6a5c22ce..e3173049931d 100644 --- a/python/tvm/tir/schedule/trace.py +++ b/python/tvm/tir/schedule/trace.py @@ -262,7 +262,7 @@ def apply_json_to_schedule(json_obj: JSON_TYPE, sch: "Schedule") -> None: """ _ffi_api.TraceApplyJSONToSchedule(json_obj, sch) # type: ignore # pylint: disable=no-member - def show(self, style: Optional[str] = None, black_format: bool = True) -> None: + def show(self, style: Optional[str] = None, black_format: bool = False) -> None: """A sugar for print highlighted TVM script. Parameters @@ -274,7 +274,7 @@ def show(self, style: Optional[str] = None, black_format: bool = True) -> None: black_format: bool - If true (default), use the formatter Black to format the TVMScript + If true, use the formatter Black to format the TVMScript """ from tvm.script.highlight import ( # pylint: disable=import-outside-toplevel cprint,