Try to fix colab encoding error

2024-03-01 10:03:02 +09:00
parent c86def5fdf
commit 7e4108add5
1 changed files with 3 additions and 1 deletions
--- a/common/stdout_wrapper.py
+++ b/common/stdout_wrapper.py
@@ -8,7 +8,9 @@ import tempfile

 class StdoutWrapper:
    def __init__(self):
-        self.temp_file = tempfile.NamedTemporaryFile(mode="w+", delete=False)
+        self.temp_file = tempfile.NamedTemporaryFile(
+            mode="w+", delete=False, encoding="utf-8"
+        )
        self.original_stdout = sys.stdout

    def write(self, message: str):