fix: error when precision_score return float

#2
by Cesaryuan - opened
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)}