Replace get_stdout() with SAFE_STDOUT

This commit is contained in:
litagin02
2023-12-30 18:45:04 +09:00
parent 6e8bcc2219
commit a9652979cc
10 changed files with 24 additions and 24 deletions

View File

@@ -26,9 +26,9 @@ class StdoutWrapper:
return self.temp_file.fileno()
def get_stdout():
# Colab 環境をチェックする
if "google.colab" in sys.modules:
return StdoutWrapper()
else:
return sys.stdout
try:
import google.colab
SAFE_STDOUT = StdoutWrapper()
except ImportError:
SAFE_STDOUT = sys.stdout