diff --git a/inference/core/interfaces/camera/utils.py b/inference/core/interfaces/camera/utils.py index ccbf2cf0ab..a11e8cfe0b 100644 --- a/inference/core/interfaces/camera/utils.py +++ b/inference/core/interfaces/camera/utils.py @@ -139,6 +139,7 @@ def __init__( self._ended_sources: Set[int] = set() self._threads_to_join: Set[int] = set() self._last_batch_yielded_time = datetime.now() + self._all_sources_len = len(self._video_sources.all_sources) def retrieve_frames_from_sources( self, @@ -175,7 +176,7 @@ def retrieve_frames_from_sources( return batch_frames def all_sources_ended(self) -> bool: - return len(self._ended_sources) >= len(self._video_sources.all_sources) + return len(self._ended_sources) >= self._all_sources_len def join_all_reconnection_threads(self, include_not_finished: bool = False) -> None: for source_ord in copy(self._threads_to_join):