search

Unbabel / COMET

A Neural Framework for MT Evaluation
https://unbabel.github.io/COMET/html/index.html
Apache License 2.0
441 stars 72 forks source link

Update base.py Fixes: AttributeError: 'dict' object has no attribute 'scores' #185

Closed ymoslem closed 7 months ago

ymoslem commented 7 months ago

Fixes: AttributeError: 'dict' object has no attribute 'scores' #183

Changed:

scores = torch.cat([pred.scores for pred in predictions], dim=0).tolist()

to:

scores = torch.cat([pred["scores"] for pred in predictions], dim=0).tolist()
mjpost commented 6 months ago

Hi @ricardorei, can you do a 2.2.1 release with this fix?