Reset stdout and stderr, also change logger stream handler.
()
| 75 | |
| 76 | |
| 77 | def reset_std(): |
| 78 | """Reset stdout and stderr, also change logger stream handler.""" |
| 79 | sys.stdout.close() |
| 80 | sys.stdout = old_stdout |
| 81 | sys.stderr = old_stderr |
| 82 | # change logger stream handler as well |
| 83 | logger = get_logger() |
| 84 | for h in logger.handlers: |
| 85 | if isinstance(h, logging.StreamHandler): |
| 86 | h.stream = sys.stdout |
| 87 | # special treat for icl_gen_inferencer logger |
| 88 | gen_logger = logging.getLogger( |
| 89 | 'opencompass.openicl.icl_inferencer.icl_gen_inferencer') |
| 90 | for h in gen_logger.handlers: |
| 91 | if isinstance(h, logging.StreamHandler): |
| 92 | h.stream = sys.stdout |
| 93 | |
| 94 | |
| 95 | def launch(task: BaseTask, tokens: SyncManager.Semaphore): |