Switched to ThreadPool
This commit is contained in:
parent
e62bb932c6
commit
6d553da093
1 changed files with 2 additions and 2 deletions
|
@ -1,7 +1,7 @@
|
||||||
import argparse
|
import argparse
|
||||||
import configparser
|
import configparser
|
||||||
import logging
|
import logging
|
||||||
from multiprocessing.pool import Pool
|
from multiprocessing.pool import ThreadPool
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
@ -39,7 +39,7 @@ var = {
|
||||||
}
|
}
|
||||||
|
|
||||||
inp.mkdir(exist_ok=True)
|
inp.mkdir(exist_ok=True)
|
||||||
with Pool() as pool:
|
with ThreadPool() as pool:
|
||||||
log.info("Converting all data")
|
log.info("Converting all data")
|
||||||
pool.map(
|
pool.map(
|
||||||
lambda x: np.save(
|
lambda x: np.save(
|
||||||
|
|
Loading…
Reference in a new issue