Fixed multiproc import
This commit is contained in:
parent
64e5cac36f
commit
2acb5f9f17
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 import ThreadPool
|
from multiprocessing.pool import Pool
|
||||||
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 ThreadPool() as pool:
|
with Pool() as pool:
|
||||||
pool.map(
|
pool.map(
|
||||||
lambda name, f: np.save(
|
lambda name, f: np.save(
|
||||||
inp.joinpath(name), f(sws_out.joinpath(name).with_suffix(".dat"))
|
inp.joinpath(name), f(sws_out.joinpath(name).with_suffix(".dat"))
|
||||||
|
|
Loading…
Reference in a new issue