@@ -14,7 +14,7 @@ class RougeScore(SingleTurnMetric):
1414 _required_columns : t .Dict [MetricType , t .Set [str ]] = field (
1515 default_factory = lambda : {MetricType .SINGLE_TURN : {"reference" , "response" }}
1616 )
17- rogue_type : t .Literal ["rouge1" , "rougeL" ] = "rougeL"
17+ rouge_type : t .Literal ["rouge1" , "rougeL" ] = "rougeL"
1818 measure_type : t .Literal ["fmeasure" , "precision" , "recall" ] = "fmeasure"
1919
2020 def __post_init__ (self ):
@@ -34,9 +34,9 @@ async def _single_turn_ascore(
3434 ) -> float :
3535 assert isinstance (sample .reference , str ), "Sample reference must be a string"
3636 assert isinstance (sample .response , str ), "Sample response must be a string"
37- scorer = self .rouge_scorer .RougeScorer ([self .rogue_type ], use_stemmer = True )
37+ scorer = self .rouge_scorer .RougeScorer ([self .rouge_type ], use_stemmer = True )
3838 scores = scorer .score (sample .reference , sample .response )
39- return getattr (scores [self .rogue_type ], self .measure_type )
39+ return getattr (scores [self .rouge_type ], self .measure_type )
4040
4141 async def _ascore (self , row : t .Dict , callbacks : Callbacks ) -> float :
4242 return await self ._single_turn_ascore (SingleTurnSample (** row ), callbacks )
0 commit comments