diff --git a/openjudge/generator/llm_grader_generator.py b/openjudge/generator/llm_grader_generator.py index afaaacbd..9be3b362 100644 --- a/openjudge/generator/llm_grader_generator.py +++ b/openjudge/generator/llm_grader_generator.py @@ -100,6 +100,7 @@ async def generate(self, dataset: List[dict], **kwargs) -> LLMGrader: rubrics = await self._generate_rubrics(dataset, **kwargs) return LLMGrader( model=self.config.model, # type: ignore + name=self.config.grader_name, mode=self.config.grader_mode, # type: ignore template=self.config.custom_evaluation_prompt, # type: ignore rubrics=rubrics,