format

Author: ji-huazhong

verl/workers/rollout/schemas.py

@@ -181,8 +181,8 @@ def initialize_request(cls, values):
                 # Only log the warning to avoid truncating in the middle of generation prompt. Consider raising an
                 # error for this case in the future.
                 # Ensure batch_data_id exists with default value if not provided
-                if 'batch_data_id' not in values:
-                    values['batch_data_id'] = cls.model_fields['batch_data_id'].default
+                if "batch_data_id" not in values:
+                    values["batch_data_id"] = cls.model_fields["batch_data_id"].default
                 logger.warning(
                     f"Prompt {values['batch_data_id']} has length {values['input_ids'].shape[-1]} "
                     f"which is greater than max_prompt_len {max_prompt_len} after applied chat template with tools."