| #!/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() | |