Spaces:
Running
Running
fix: error when precision_score return float
Browse filesWhen precision_score returns float, score.size will throw an error `AttributeError: 'float' object has no attribute 'size'`
- 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":
|
|
|
142 |
sample_weight=sample_weight,
|
143 |
zero_division=zero_division,
|
144 |
)
|
145 |
+
return {"precision": score if isinstance(score, float) else float(score)}
|