def _set_signature_columns_if_needed()

in src/open-r1-multimodal/src/open_r1/trainer/grpo_trainer.py [0:0]

• 3 lines of code
• 2 McCabe index (conditional complexity)

    def _set_signature_columns_if_needed(self):
        # If `self.args.remove_unused_columns` is True, non-signature columns are removed.
        # By default, this method sets `self._signature_columns` to the model's expected inputs.
        # In GRPOTrainer, we preprocess data, so using the model's signature columns doesn't work.
        # Instead, we set them to the columns expected by the `training_step` method, hence the override.
        if self._signature_columns is None:
            self._signature_columns = ["prompt"]