def find_nearest_matches(x, y):
num_processes = CONFIG.get("num_processes")
if num_processes > 1:
pool = multiprocessing.Pool(processes=num_processes)
batch_size = int(math.ceil(len(y) / num_processes))
parts = [(x, y[i:i + batch_size])
for i in range(0, len(y), batch_size)]
res = pool.map(_ff, parts)
res_par = np.hstack(res)
return res_par
return _find_nearest_matches(x, y)
def find_nearest_matches(x, y):
num_processes = CONFIG.get("num_processes")
if num_processes > 1:
pool = multiprocessing.Pool(processes=num_processes)
batch_size = int(math.ceil(len(y) / num_processes))
parts = [(x, y[i:i + batch_size]) for i in range(0, len(y),
batch_size)]
res = pool.map(_ff, parts)
res_par = np.hstack(res)
return res_par
return _find_nearest_matches(x, y)
def _ff(a):
return _find_nearest_matches(*a)
def _ff(a):
return _find_nearest_matches(*a)