Wait for thread pool executor to finish python