def MyParallelPool(nodes=None): if nodes is None or nodes > 1: p = ParallelPool(nodes) try: yield p finally: p.close() p.join() p.clear() else: #print("Using PseudoPool!") yield PseudoPool()