Add special resolution for / in src of container mounts for slurm

Signed-off-by: Hemil Desai <[email protected]>

Author: hemildesai

src/nemo_run/core/execution/slurm.py

@@ -865,6 +865,10 @@ def get_container_flags(
             _container_flags = ["--container-image", container_image] if container_image else []
 
             new_mounts = copy.deepcopy(base_mounts)
+            for i, mount in enumerate(new_mounts):
+                if mount.startswith("/$nemo_run"):
+                    new_mounts[i] = mount.replace("$nemo_run", src_job_dir)
+
             new_mounts.append(f"{src_job_dir}:/{RUNDIR_NAME}")
             _mount_arg = ",".join(new_mounts)
             _container_flags += ["--container-mounts", _mount_arg]