File size: 445 Bytes
83adb51 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | #!/usr/bin/env python3
"""Wrapper that ensures flushing for background warmup."""
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
# Monkey-patch print to flush after every call
_original_print = print
def _flushing_print(*args, **kwargs):
kwargs.setdefault("flush", True)
_original_print(*args, **kwargs)
import builtins
builtins.print = _flushing_print
from warmup import warmup
warmup()
|