liqian 1 year ago
parent
commit
95b7aa156e
1 changed files with 4 additions and 2 deletions
  1. 4 2
      video_recall.py

+ 4 - 2
video_recall.py

@@ -2105,11 +2105,13 @@ class PoolRecall(object):
             # 补充不区分地域小时级结果(不做离线去重)
             recall_h_h_result = self.recall_h(province_code=region_code, size=size, expire_time=expire_time)
             # 去重
+            recall_result = recall_result[:recall_num]
+            recall_video_ids = [video.get('videoId') for video in recall_result]
             for video in recall_h_h_result:
                 video_id = video.get('videoId')
-                if video_id not in now_video_ids:
+                if video_id not in recall_video_ids:
                     recall_result.append(video)
-                    now_video_ids.append(video_id)
+                    recall_video_ids.append(video_id)
             return recall_result
 
     def rov_pool_recall_with_region(self, size=4, expire_time=24*3600):