From 036a9564732c274cc630ef245f7c03e73e409fff Mon Sep 17 00:00:00 2001
From: SkalskiP <piotr.skalski92@gmail.com>
Date: Sat, 26 Jul 2025 17:46:04 +0200
Subject: [PATCH] fix for #1911

---
 examples/time_in_zone/requirements.txt                 | 4 +++-
 examples/time_in_zone/scripts/download_from_youtube.py | 2 +-
 supervision/annotators/utils.py                        | 6 +++++-
 test/annotators/test_utils.py                          | 2 +-
 4 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/examples/time_in_zone/requirements.txt b/examples/time_in_zone/requirements.txt
index 9f9446c8..b5ff1911 100644
--- a/examples/time_in_zone/requirements.txt
+++ b/examples/time_in_zone/requirements.txt
@@ -1,4 +1,6 @@
 supervision
 ultralytics
 inference
-pytube
+# https://github.com/pytube/pytube/issues/2044
+# pytube
+pytubefix
diff --git a/examples/time_in_zone/scripts/download_from_youtube.py b/examples/time_in_zone/scripts/download_from_youtube.py
index d8673631..b7400955 100644
--- a/examples/time_in_zone/scripts/download_from_youtube.py
+++ b/examples/time_in_zone/scripts/download_from_youtube.py
@@ -3,7 +3,7 @@ from __future__ import annotations
 import argparse
 import os
 
-from pytube import YouTube
+from pytubefix import YouTube
 
 
 def main(url: str, output_path: str | None, file_name: str | None) -> None:
diff --git a/supervision/annotators/utils.py b/supervision/annotators/utils.py
index a84e2049..511c68de 100644
--- a/supervision/annotators/utils.py
+++ b/supervision/annotators/utils.py
@@ -142,7 +142,11 @@ def resolve_color(
         detection_idx=detection_idx,
         color_lookup=color_lookup,
     )
-    if color_lookup == ColorLookup.TRACK and idx == PENDING_TRACK_ID:
+    if (
+        isinstance(color_lookup, ColorLookup)
+        and color_lookup == ColorLookup.TRACK
+        and idx == PENDING_TRACK_ID
+    ):
         return PENDING_TRACK_COLOR
     return get_color_by_index(color=color, idx=idx)
 
diff --git a/test/annotators/test_utils.py b/test/annotators/test_utils.py
index 2fdccec1..d6abc381 100644
--- a/test/annotators/test_utils.py
+++ b/test/annotators/test_utils.py
@@ -97,7 +97,7 @@ from test.test_utils import mock_detections
 def test_resolve_color_idx(
     detections: Detections,
     detection_idx: int,
-    color_lookup: ColorLookup,
+    color_lookup: ColorLookup | np.ndarray,
     expected_result: int | None,
     exception: Exception,
 ) -> None: