lvwerra HF staff commited on
Commit
2b579f6
·
1 Parent(s): 088d973

Update Space (evaluate main: 5aa3982a)

Browse files
Files changed (2) hide show
  1. f1.py +1 -1
  2. requirements.txt +1 -1
f1.py CHANGED
@@ -127,4 +127,4 @@ class F1(evaluate.Metric):
127
  score = f1_score(
128
  references, predictions, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
129
  )
130
- return {"f1": float(score) if score.size == 1 else score}
 
127
  score = f1_score(
128
  references, predictions, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
129
  )
130
+ return {"f1": score if getattr(score, "size", 1) > 1 else float(score)}
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- git+https://github.com/huggingface/evaluate@55f1bc6e072b05c2d9db1589a07e20f38902b1ec
2
  scikit-learn
 
1
+ git+https://github.com/huggingface/evaluate@5aa3982a9a8c86e506860e381d428a64b0cce73b
2
  scikit-learn