diff --git a/nemo_run/core/execution/skypilot.py b/nemo_run/core/execution/skypilot.py index 9e0d01a9..0fe2fa82 100644 --- a/nemo_run/core/execution/skypilot.py +++ b/nemo_run/core/execution/skypilot.py @@ -374,7 +374,7 @@ def to_task( ) file_mounts = self.file_mounts or {} file_mounts["/nemo_run"] = self.job_dir - task.set_file_mounts(self.file_mounts) + task.set_file_mounts(file_mounts) task.set_resources(self.to_resources()) if env_vars: