commit c1ef21404639757fd22ad3d2e030b059ba9234ef
author Sherlock Huang <bahuang@meta.com> Tue Jul 23 16:41:55 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Jul 23 16:41:55 2024 +0000
tree f4a30ed18ad380bfbc7c5cb172c3ea55798e5094
parent db376fb6434a18d84b250b987ecf3b1a4bcd0586

Print ExportedProgram without color by default (#131399)

Summary:
Without plugin, colored ExportedProgram is not really readable.

![image](https://github.com/user-attachments/assets/319920a9-bb4b-4ad2-bcac-0c4f76973b11)

Test Plan: CI

Differential Revision: D60074481

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131399
Approved by: https://github.com/angelayi

torch/export/exported_program.py

diff --git a/torch/export/exported_program.py b/torch/export/exported_program.py
index 3de0256..15778bd 100644
--- a/torch/export/exported_program.py
+++ b/torch/export/exported_program.py
@@ -941,7 +941,7 @@
 
     def __str__(self) -> str:
         graph_module = self.graph_module.print_readable(
-            print_output=False, colored=True
+            print_output=False, colored=False
         ).replace("\n", "\n    ")
         string = (
             "ExportedProgram:\n"