Quick test for windows

huggingface · Dec 25, 2023 · 24f11b6 · 24f11b6
1 parent 8ccf967
commit 24f11b6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py
@@ -40,7 +40,9 @@
         >>> confusion_matrix_metric = evaluate.load("confusion_matrix")
         >>> results = confusion_matrix_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0])
         >>> print(results)  # doctest: +NORMALIZE_WHITESPACE
-        {'confusion_matrix': array([[1, 0, 1], [0, 2, 0], [1, 1, 0]])}
+        {'confusion_matrix': array([[1, 0, 1],
+                            [0, 2, 0],
+                            [1, 1, 0]], dtype=int64)}
 """