@@ -663,6 +663,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, flowConfig=None):
all_recall_rank = rov_recall_rank + flow_recall_rank
if flow_num > 0:
rank_result.extend(all_recall_rank[:size - top_K])
+ return rank_result, flow_num
else:
i = 0
while i < size - top_K: