如何调用其他py文件函数的返回值？

这是我想调用的infer.py中的函数：

def _predict_ids_to_seq(predict_ids, id2word, beam_size):
    predicts = []
    for single_predict in predict_ids:
        for i in range(beam_size):
            predict_list = np.ndarray.tolist(single_predict[:, :, i])  #一维全取，二维全取，三维取i值
            predict_seq = [id2word[idx] for idx in predict_list[0] 
                if idx in id2word if idx >=4]
            predicts.append("".join(predict_seq))
    return predicts

我想将predicts列表赋给另一个py文件中的变量probs_temp_int

我是这样调用的：

from seq2seq_dialog.infer import _predict_ids_to_seq
...
probs_temp_int = _predict_ids_to_seq(predict_ids, id2word, beam_size)

但是由于 _predict_ids_to_seq函数的三个变量还需其他计算才能得到，所以不能给定值，求助如何调用这种函数的返回值？

再定义成函数，把不能确定的变量作为参数