rename _ThrottleIterator -> _ThrottleIteratorMixin

ebonnal · Jan 2, 2025 · 0714185 · 0714185
1 parent 25f2407
commit 0714185
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/streamable/iterators.py b/streamable/iterators.py
@@ -374,7 +374,7 @@ def __next__(self) -> T:
                 self._log()
 
 
-class _ThrottleIterator(Iterator[T]):
+class _ThrottleIteratorMixin(Generic[T]):
     def __init__(self, iterator: Iterator[T]) -> None:
         validate_iterator(iterator)
         self.iterator = iterator
@@ -388,7 +388,7 @@ def safe_next(self) -> Tuple[Optional[T], Optional[Exception]]:
             return None, e
 
 
-class IntervalThrottleIterator(_ThrottleIterator[T]):
+class IntervalThrottleIterator(_ThrottleIteratorMixin[T], Iterator[T]):
     def __init__(self, iterator: Iterator[T], interval: datetime.timedelta) -> None:
         super().__init__(iterator)
         validate_throttle_interval(interval)
@@ -409,7 +409,7 @@ def __next__(self) -> T:
         return cast(T, elem)
 
 
-class YieldsPerPeriodThrottleIterator(_ThrottleIterator[T]):
+class YieldsPerPeriodThrottleIterator(_ThrottleIteratorMixin[T], Iterator[T]):
     def __init__(
         self,
         iterator: Iterator[T],