From 3b2a229ba4be3dcb4751910815f787bd01fd4d88 Mon Sep 17 00:00:00 2001
From: Omar Sanseviero <osanseviero@gmail.com>
Date: Mon, 25 Dec 2023 10:56:56 +0100
Subject: [PATCH] Update confusion_matrix.py

---
 metrics/confusion_matrix/confusion_matrix.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py
index 37e0f78f..54a1914d 100644
--- a/metrics/confusion_matrix/confusion_matrix.py
+++ b/metrics/confusion_matrix/confusion_matrix.py
@@ -42,7 +42,7 @@
         >>> print(results)  # doctest: +NORMALIZE_WHITESPACE
         {'confusion_matrix': array([[1, 0, 1],
                             [0, 2, 0],
-                            [1, 1, 0]], dtype=int64)}
+                            [1, 1, 0]])}
 """