multiprocessing.pool.RemoteTraceback

[Dimear]

multiprocessing.pool.RemoteTraceback:
"""
Traceback (most recent call last):
File "D:\study\PYTHON\lib\multiprocessing\pool.py", line 125, in worker
result = (True, func(*args, **kwds))
File "D:\study\PYTHON\lib\multiprocessing\pool.py", line 51, in starmapstar
return list(itertools.starmap(args[0], args[1]))
File "E:\python\HID-master\CDK.py", line 37, in perform_prediction_one_cascade
p_cascade.append(idx_2_user[predicted_cascade_user_index[i]])
KeyError: 838
"""

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "E:/python/HID-master/test.py", line 87, in
test(config=config, save_dir=config['output_path'], test_epoch=config['test_epoch'], embedding_scale=config['embedding_scale'], corpus_test=corpus_test)
File "E:\python\HID-master\CDK.py", line 363, in test
diffpath_user_predicted = perform_prediction_multiprocessing(
File "E:\python\HID-master\CDK.py", line 62, in perform_prediction_multiprocessing
predicted_cascades = pool.starmap(perform_prediction_one_cascade, list_of_args_tuple)
File "D:\study\PYTHON\lib\multiprocessing\pool.py", line 372, in starmap
return self._map_async(func, iterable, starmapstar, chunksize).get()
File "D:\study\PYTHON\lib\multiprocessing\pool.py", line 768, in get
raise self._value
KeyError: 838

hi,I encountered the above problems when I reproduced the code. Can you help me solve them? Thanks.