Fix tests

2024-10-22 20:14:06 +02:00
parent ee52b8b782
commit b46db7ea73
2 changed files with 58 additions and 55 deletions
--- a/lerobot/common/datasets/image_writer.py
+++ b/lerobot/common/datasets/image_writer.py
@@ -120,7 +120,7 @@ class ImageWriter:
                wait(self.futures, timeout=timeout)
                progress_bar.update(len(self.futures))
        else:
-            self._stop_processes(self.processes, self.image_queue, timeout)
+            self._stop_processes(timeout)

    def _stop_processes(self, timeout) -> None:
        for _ in self.processes: