Error when translating with scores (--with_scores) option enabled

[ncicio]

When attempting to translate with the --with_scores option enabled, an error is encountered in the OpenNMT translation module. The specific error message is:

AttributeError: 'float' object has no attribute 'item'

This error arises from the translator.py file at line 437, within a list comprehension where it attempts to access the item() attribute of a float object in trans.pred_scores.

Expected Behavior: The translation process with scores should proceed without encountering any errors, providing the translated text along with corresponding scores.

Actual Behavior: The error mentioned above is encountered, halting the translation process and preventing the retrieval of translation scores.

Additional Context:

This issue occurs specifically when translating with scores enabled; without scores, the translation process functions as expected. The error seems to stem from an attempt to access the item() attribute of a float object, which is invalid. Possible Solution: Inspect the code within the translator.py file at line 437 to identify the source of the error. Ensure that the trans.pred_scores attribute is properly handled when the --with_scores option is enabled.

Environment:

Python version: 3.9 OpenNMT version: 3.5.1 Operating System: Ubuntu

[vince62s]

you need to remove the .item() can you send a PR?