requested changes

Author: justusschock

delira/data_loading/data_loader.py

@@ -1,6 +1,5 @@
 import numpy as np
 from batchgenerators.dataloading.data_loader import SlimDataLoaderBase
-from multiprocessing import Queue
 from queue import Empty
 import logging
 
@@ -26,7 +25,7 @@ def __init__(self, dataset: AbstractDataset,
             dataset to perform sample loading
         batch_size : int
             number of samples per batch
-        sampler_queues : list :class:`multiprocessing.Queue`
+        sampler_queues : list of :class:`multiprocessing.Queue`
             the queue,s the sample indices to load will be put to.
             Necessary for interprocess communication
         num_batches : int
@@ -81,8 +80,6 @@ def generate_train_batch(self):
             try:
                 idxs = sampler_queue.get(timeout=0.2)
 
-                # idxs = self.sampler_queue.get()
-
                 result = [self._get_sample(_idx) for _idx in idxs]
 
                 result_dict = {}

delira/data_loading/data_manager.py

@@ -44,7 +44,7 @@ def __init__(self, data_loader: BaseDataLoader, transforms,
             the sampler to use; must be used here instead of inside the
             dataloader to avoid duplications and oversampling due to
             multiprocessing
-        sampler_queues : list of g:class:`multiprocessing.Queue`
+        sampler_queues : list of :class:`multiprocessing.Queue`
             queues to pass the sample indices to the actual dataloader
         num_cached_per_queue : int
             the number of samples to cache per queue (only necessary if not in