ModuleNotFoundError: No module named 'cray_infra.training.cancel

[TheAmazingRoderic]

New issue introduced with Commit c72e4bc when running ./scalarlm up nvidia

...
cray-nvidia-1 | (APIServer pid=471) File "/app/cray/infra/cray_infra/api/fastapi/routers/megatron_router.py", line 19, in
cray-nvidia-1 | (APIServer pid=471) from cray_infra.training.cancel import cancel
cray-nvidia-1 | (APIServer pid=471) ModuleNotFoundError: No module named 'cray_infra.training.cancel'
...

commenting out line 19 allowed successful running of ./scalarlm up