Cesaryuan commited on
Commit
cf1046d
·
verified ·
1 Parent(s): e45c3f7

fix: error when precision_score return float

Browse files

When precision_score returns float, score.size will throw an error `AttributeError: 'float' object has no attribute 'size'`

Files changed (1) hide show
  1. precision.py +1 -1
precision.py CHANGED
@@ -142,4 +142,4 @@ class Precision(evaluate.Metric):
142
  sample_weight=sample_weight,
143
  zero_division=zero_division,
144
  )
145
- return {"precision": float(score) if score.size == 1 else score}
 
142
  sample_weight=sample_weight,
143
  zero_division=zero_division,
144
  )
145
+ return {"precision": score if isinstance(score, float) else float(score)}