fix GeneratorDataset time out

5c9b0657 · yanghaitao · efb12d5e · 5c9b0657
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/dataset/engine/datasets.py mindspore/dataset/engine/datasets.py +1 -1

未找到文件。
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -3118,7 +3118,7 @@ class _GeneratorWorker(multiprocessing.Process):
        """
        Get function for worker result queue. Block with timeout.
        """
-        return self.res_queue.get(timeout=5)
+        return self.res_queue.get()

    def __del__(self):
        self.terminate()