(hq_env: HqEnv, count=1000, nodes=1)
| 78 | |
| 79 | |
| 80 | def prepare_tasks(hq_env: HqEnv, count=1000, nodes=1): |
| 81 | args = ["submit", f"--array=0-{count}"] |
| 82 | if nodes > 1: |
| 83 | args += ["--nodes", str(nodes)] |
| 84 | args += ["sleep", "1"] |
| 85 | hq_env.command(args) |
| 86 | |
| 87 | |
| 88 | def remove_queue(hq_env: HqEnv, queue_id: int, force=False, **kwargs): |