fix numbering bug

Author: bens

fuse/eval/metrics/libs/classification.py

@@ -186,7 +186,7 @@ def auc_pr(
             y_score = np.asarray(pred)[:, pos_class_index]
 
         precision, recall, _ = metrics.precision_recall_curve(
-            probas_pred=y_score,
+            y_score=y_score,
             y_true=np.asarray(target) == pos_class_index,
             sample_weight=sample_weight,
         )

fuse/eval/metrics/metrics_common.py

@@ -190,7 +190,10 @@ def collect(self, batch: Dict) -> None:
                 if isinstance(value, np.ndarray) and value.ndim == 0:
                     self._collected_data[name].append(value.item())
                 else:
-                    self._collected_data[name].extend(value)
+                    if isinstance(value, np.ndarray):
+                        self._collected_data[name].extend(value.tolist())
+                    else:
+                        self._collected_data[name].extend(value)
 
         # extract ids and store it in self._collected_ids
         if self._to_collect_ids: