分配了Slurm作业后在主机节点上进行回调

Question

我想依次做两件事：

1. 通过sbatch提交工作
2. 已分配作业后，检索分配的节点的hostname，并使用该名称在 host （登录）节点上执行第二条命令。

步骤2是困难的部分。我想我可以编写一个轮询squeue的Python脚本。有没有更好的办法？我是否可以设置一个回调，Slurm将在作业开始运行后执行该回调？

（如果您感到好奇，我的动机是在计算节点上启动Jupyter笔记本，并像this blog post一样自动设置ssh端口转发。）