Remove timeout on wait for completed

Author: jeromekelleher

bio2zarr/core.py

@@ -223,7 +223,7 @@ def submit(self, *args, **kwargs):
         self.futures.add(future)
         return future
 
-    def wait_for_completed(self, timeout):
+    def wait_for_completed(self, timeout=None):
         done, not_done = cf.wait(self.futures, timeout, cf.FIRST_COMPLETED)
         for future in done:
             exception = future.exception()

bio2zarr/vcf.py

@@ -1595,7 +1595,7 @@ def encode(
                         f"Memory budget {display_size(max_memory)} exceeded: "
                         f"used={display_size(used_memory)} needed={display_size(wp.memory)}"
                     )
-                    futures = pwm.wait_for_completed(timeout=5)
+                    futures = pwm.wait_for_completed()
                     released_mem = sum(
                         future_to_memory_use.pop(future) for future in futures
                     )