Skip to content

Commit ce90faa

Browse files
committed
fix: remove preset_reward_function from RLVR hyperparameters to prevent empty value being sent to service
1 parent addb37b commit ce90faa

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

‎sagemaker-train/src/sagemaker/train/rlvr_trainer.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,9 @@ def _process_hyperparameters(self):
171171
if hasattr(self.hyperparameters, 'reward_lambda_arn'):
172172
delattr(self.hyperparameters, 'reward_lambda_arn')
173173
self.hyperparameters._specs.pop('reward_lambda_arn', None)
174+
if hasattr(self.hyperparameters, 'preset_reward_function'):
175+
delattr(self.hyperparameters, 'preset_reward_function')
176+
self.hyperparameters._specs.pop('preset_reward_function', None)
174177
if hasattr(self.hyperparameters, 'data_path'):
175178
delattr(self.hyperparameters, 'data_path')
176179
self.hyperparameters._specs.pop('data_path', None)

0 commit comments

Comments
 (0)