refactor(AutoData): optionally provide workers as kwarg to fetch

g5becks · Oct 27, 2022 · 7c41709 · 7c41709
1 parent ed2a5be
commit 7c41709
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/autotrader/autodata.py b/autotrader/autodata.py
@@ -234,7 +234,8 @@ def fetch(
         """
         func = getattr(self, f"_{self._feed}")
         if isinstance(instrument, list):
-            with ThreadPoolExecutor() as executor:
+            max_workers = kwargs["workers"] if "workers" in kwargs else None
+            with ThreadPoolExecutor(max_workers=max_workers) as executor:
                 futures = {}
                 for i in instrument:
                     futures[i] = executor.submit(