Skip to content

Commit 83c6949

Browse files
Merge pull request #221 from TaekyungHeo/nemo-abs
Use absolute paths in NeMo launcher
2 parents 0aed180 + f533061 commit 83c6949

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/cloudai/schema/test_template/nemo_launcher/slurm_command_gen_strategy.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,14 +40,15 @@ def gen_exec_command(
4040
self.system.install_path
4141
/ NeMoLauncherSlurmInstallStrategy.SUBDIR_PATH
4242
/ NeMoLauncherSlurmInstallStrategy.REPOSITORY_NAME
43-
)
43+
).absolute()
44+
output_path_abs = output_path.absolute()
4445
overriden_cmd_args = self._override_cmd_args(self.default_cmd_args, cmd_args)
4546
self.final_cmd_args = {
46-
k: self._handle_special_keys(k, v, str(launcher_path), str(output_path))
47+
k: self._handle_special_keys(k, v, str(launcher_path), str(output_path_abs))
4748
for k, v in overriden_cmd_args.items()
4849
}
49-
self.final_cmd_args["base_results_dir"] = str(output_path)
50-
self.final_cmd_args["training.model.data.index_mapping_dir"] = str(output_path)
50+
self.final_cmd_args["base_results_dir"] = str(output_path_abs)
51+
self.final_cmd_args["training.model.data.index_mapping_dir"] = str(output_path_abs)
5152
self.final_cmd_args["launcher_scripts_path"] = str(launcher_path / "launcher_scripts")
5253

5354
for key, value in final_env_vars.items():

0 commit comments

Comments
 (0)