How to run SlurmCluster object instead of my own flimsy slurm scripts

Hey all! I’m finding lightning massively boosting my productivity, thanks to everyone involved! I’m running on a cluster with Slurm. I would like to use the SlurmCluster object instead of my own flimsy slurm scripts. The problem is that I need to execute the experiment on a singularity (very similar to docker) container. My current slurm submission script runs train.py by: singularity exec --nv image.simg python train.py --<options> . Can this be handled by SlurmCluster neatly? I havent found any resources on this. Thanks!