| 1 | """A seat of a kaggle_environments game for a bot in its own process: env.run gets a function that passes each |
| 2 | observation to the bot's process and its answer back. A late or broken answer is the game's no-op.""" |
| 3 | import json, os, sys |
| 4 | |
| 5 | HERE = os.path.dirname(os.path.abspath(__file__)) |
| 6 | sys.path.insert(0, os.path.join(os.path.dirname(HERE), 'kit')) |
| 7 | from host import Child |
| 8 | from rule_agent import public_config |
| 9 | |
| 10 | |
| 11 | def seat(folder, limit=1.0, idle=None, start=30): |
| 12 | main = os.path.join(folder, 'main.py') |
| 13 | if not os.path.isfile(main): |
| 14 | sys.exit(f'No bot in {folder}: add main.py with agent(obs, config), as on Kaggle') |
| 15 | child = Child([sys.executable, '-s', os.path.join(HERE, 'kaggle_child.py'), main], cwd=folder, limit=limit, start=start) |
| 16 | |
| 17 | def agent(obs, config): |
| 18 | answer = child.ask({'obs': json.loads(json.dumps(obs, default=str)), |
| 19 | 'config': public_config(json.loads(json.dumps(config, default=str)))}) |
| 20 | return idle if answer is None else answer |
| 21 | |
| 22 | agent.child = child |
| 23 | return agent |
