add cal_topk.py comments