rules, serialization and processing

1 files changed, 2 insertions, 1 deletions

diff --git a/sec_certs/parallel_processing.py b/sec_certs/parallel_processing.py
index 20a82f32..3b91f652 100644
--- a/sec_certs/parallel_processing.py
+++ b/sec_certs/parallel_processing.py
@@ -1,12 +1,13 @@
 from tqdm import tqdm
 from multiprocessing.pool import Pool, ThreadPool
-from typing import Callable, Iterable, Optional
+from typing import Callable, Iterable, Optional, Union
 import time
 
 
 def process_parallel(func: Callable, items: Iterable, max_workers: int, callback: Optional[Callable] = None,
                      use_threading: bool = True, progress_bar: bool = True, unpack: bool = False,
                      progress_bar_desc: Optional[str] = None):
+    pool: Union[Pool, ThreadPool]
     if use_threading is True:
         pool = ThreadPool(max_workers)
     else: