Skip to content

Commit 0e760a2

Browse files
committed
Rename nemo config back to nemo_config_* fix archive pattern.
1 parent 4f73321 commit 0e760a2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

var/ramble/repos/builtin/applications/py-nemo-2/application.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,13 @@ class PyNemo2(ExecutableApplication):
2929
tags("ml-framework", "machine-learning")
3030

3131
# Add Nemo 2.0 config to archive.
32-
archive_pattern("{experiment_run_dir}/{nemo_2_config_name}")
32+
archive_pattern("{experiment_run_dir}/{nemo_config_name}/*")
3333

3434
executable(
3535
"pretraining_exec",
3636
'bash -c "'
3737
"{custom_injected_string}; "
38-
'python3 -u {experiment_run_dir}/{nemo_2_config_name};"',
38+
'python3 -u {experiment_run_dir}/{nemo_config_name};"',
3939
use_mpi=True,
4040
)
4141

@@ -79,16 +79,16 @@ class PyNemo2(ExecutableApplication):
7979
)
8080

8181
workload_variable(
82-
"nemo_2_config_dir_path",
82+
"nemo_config_dir_path",
8383
default="",
8484
description="Path to dir containing Nemo 2.0 python configs to be used.",
8585
workload_group="all_workloads",
8686
)
8787

8888
workload_variable(
89-
"nemo_2_config_name",
89+
"nemo_config_name",
9090
default="",
91-
description="Name of NeMo 2.0 config under {nemo_2_config_dir_path}.",
91+
description="Name of NeMo 2.0 config under {nemo_config_dir_path}.",
9292
workload_group="all_workloads",
9393
)
9494

0 commit comments

Comments
 (0)