liqian 3 years ago
parent
commit
de8a72ab3e
1 changed files with 2 additions and 1 deletions
  1. 2 1
      video_recall.py

+ 2 - 1
video_recall.py

@@ -73,7 +73,7 @@ class PoolRecall(object):
             debug_tm_e = time.time()
             if filtered_result:
                 # 添加视频源参数 pushFrom, abCode
-                temp_result = [{'videoId': item, 'rovScore': video_score[item],
+                temp_result = [{'videoId': int(item), 'rovScore': video_score[int(item)],
                                 'pushFrom': 'recall_pool', 'abCode': self.ab_code}
                                for item in filtered_result]
                 rov_pool_recall_result.extend(temp_result)
@@ -172,6 +172,7 @@ class PoolRecall(object):
         videos = []
         check_result = []
         for video_id in video_ids:
+            video_id = int(video_id)
             for flow_pool in flow_pool_mapping[video_id]:
                 # 判断是否有本地分发记录
                 cur_count = get_videos_local_distribute_count(video_id=video_id, flow_pool=flow_pool)