diff --git a/src/tqdm_publisher/_subscriber.py b/src/tqdm_publisher/_subscriber.py index 69f10a7..083b7b4 100644 --- a/src/tqdm_publisher/_subscriber.py +++ b/src/tqdm_publisher/_subscriber.py @@ -4,7 +4,7 @@ class TQDMProgressSubscriber(TQDMProgressPublisher): - def __init__(self, on_progress_update: callable, iterable: Optional[Iterable[Any]]=None, **tqdm_kwargs): + def __init__(self, on_progress_update: callable, iterable: Optional[Iterable[Any]] = None, **tqdm_kwargs): super().__init__(iterable=iterable, **tqdm_kwargs) def run_on_progress_update(format_dict: Dict[str, Any]): diff --git a/tests/test_subscriber.py b/tests/test_subscriber.py index e187ce9..a91e2a0 100644 --- a/tests/test_subscriber.py +++ b/tests/test_subscriber.py @@ -65,8 +65,8 @@ def test_callback(data): # Simulate updatse to trigger the callback for task in asyncio.as_completed(tasks): await task - subscriber.update(1) # Update by 1 iteration + subscriber.update(1) # Update by 1 iteration - assert n_callback_executions > 1 + assert n_callback_executions > 1 - subscriber.close() \ No newline at end of file + subscriber.close()