From 3e5763ebcf7e8bb0062745dda83aef22d307bf95 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Mon, 5 Apr 2021 13:50:31 +0100 Subject: [PATCH] fixes ThreadDataLoader Signed-off-by: Wenqi Li --- monai/data/thread_buffer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/data/thread_buffer.py b/monai/data/thread_buffer.py index da5f864900..8ea71e3555 100644 --- a/monai/data/thread_buffer.py +++ b/monai/data/thread_buffer.py @@ -88,7 +88,7 @@ def __init__(self, dataset: Dataset, num_workers: int = 0, **kwargs): super().__init__(dataset, num_workers, **kwargs) # ThreadBuffer will use the inherited __iter__ instead of the one defined below - self.buffer = ThreadBuffer(super()) + self.buffer = ThreadBuffer(super().__iter__()) def __iter__(self): yield from self.buffer