giulio98 commited on
Commit
6b96e22
1 Parent(s): 01353be

Update codebleu.py

Browse files
Files changed (1) hide show
  1. codebleu.py +3 -2
codebleu.py CHANGED
@@ -76,8 +76,9 @@ class CodeBLEU(evaluate.Metric):
76
  inputs_description=_KWARGS_DESCRIPTION,
77
  # This defines the format of each prediction and reference
78
  features=datasets.Features({
79
- 'predictions': datasets.Value('int64'),
80
- 'references': datasets.Value('int64'),
 
81
  }),
82
  # Homepage of the module for documentation
83
  homepage="http://module.homepage",
 
76
  inputs_description=_KWARGS_DESCRIPTION,
77
  # This defines the format of each prediction and reference
78
  features=datasets.Features({
79
+ 'predictions': datasets.Value('string'),
80
+ 'references': datasets.Value('string'),
81
+ 'language': datasets.Value('string'),
82
  }),
83
  # Homepage of the module for documentation
84
  homepage="http://module.homepage",