danieldux commited on
Commit
770d3de
1 Parent(s): fb1750b

apply helper to references

Browse files
Files changed (1) hide show
  1. isco_hierarchical_accuracy.py +1 -0
isco_hierarchical_accuracy.py CHANGED
@@ -274,6 +274,7 @@ class ISCO_Hierarchical_Accuracy(evaluate.Metric):
274
  # Convert the inputs to strings
275
  if len(predictions[0]) > 4:
276
  predictions = [self._extract_isco_code(p) for p in predictions]
 
277
  predictions = [str(p) for p in predictions]
278
  references = [str(r) for r in references]
279
 
 
274
  # Convert the inputs to strings
275
  if len(predictions[0]) > 4:
276
  predictions = [self._extract_isco_code(p) for p in predictions]
277
+ references = [self._extract_isco_code(r) for r in references]
278
  predictions = [str(p) for p in predictions]
279
  references = [str(r) for r in references]
280