Skip to content

Commit c4fea25

Browse files
authored
fix: Update loss_algo attribute format in RLTestConfig (#1133)
This pull request makes a small but important update to the RL training command generation logic, ensuring that the loss algorithm parameter is properly name-spaced for downstream compatibility.
1 parent e227e61 commit c4fea25

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dags/post_training/util/test_config_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def generate_rl_training_command(
133133
f"tokenizer_path={self.tokenizer_path} "
134134
f"load_parameters_path={self.load_parameters_path} "
135135
f"base_output_directory={self.base_dir} "
136-
f"loss_algo={loss_algo.loss_name}"
136+
f"rl.loss_algo={loss_algo.loss_name}"
137137
)
138138

139139
# Return as tuple for k8s yaml compatibility.

0 commit comments

Comments
 (0)