From 950744111aa614e560bb70157b63ed52cc7e4deb Mon Sep 17 00:00:00 2001
From: kirilllzaitsev <kzaitsev88@gmail.com>
Date: Wed, 19 Jul 2023 22:54:39 +0200
Subject: [PATCH] fix type annotations

---
 supervision/metrics/detection.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/supervision/metrics/detection.py b/supervision/metrics/detection.py
index 95f2c62f..8025b1d4 100644
--- a/supervision/metrics/detection.py
+++ b/supervision/metrics/detection.py
@@ -1,7 +1,9 @@
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Callable, List, Optional, Tuple
-import matplotlib
 
+import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np
 
@@ -32,7 +34,7 @@ class ConfusionMatrix:
         conf_threshold: float,
         iou_threshold: float,
         classes: Optional[List[str]],
-    ) -> "ConfusionMatrix":
+    ) -> ConfusionMatrix:
         """
         Create ConfusionMatrix from matrix.
 
@@ -60,7 +62,7 @@ class ConfusionMatrix:
         classes: List[str],
         conf_threshold: float = 0.3,
         iou_threshold: float = 0.5,
-    ) -> "ConfusionMatrix":
+    ) -> ConfusionMatrix:
         """
         Calculate confusion matrix based on predicted and ground-truth detections.
 
@@ -219,7 +221,7 @@ class ConfusionMatrix:
         callback: Callable[[np.ndarray], sv.Detections],
         conf_threshold: float = 0.3,
         iou_threshold: float = 0.5,
-    ) -> "ConfusionMatrix":
+    ) -> ConfusionMatrix:
         """
         Create confusion matrix from dataset and callback function.