liqian 3 年之前
父節點
當前提交
056f24c2f1
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      recommend.py

+ 2 - 2
recommend.py

@@ -372,7 +372,7 @@ def video_homepage_recommend(mid, uid, size, app_type, algo_type, client_info, a
             top_K = config_.K
             flow_pool_P = config_.P
         else:
-            exp_item_id_list = [int(item.get('expItemId')) for item in ab_exp_info if item is not None]
+            exp_item_id_list = [item.get('expItemId') for item in ab_exp_info]
             if config_.EXP_ITEM_ID['rec_size'] in exp_item_id_list:
                 config_value = ab_exp_info[exp_item_id_list.index(config_.EXP_ITEM_ID['rec_size'])].get('configValue')
                 size = int(config_value.get('size', 4))
@@ -415,7 +415,7 @@ def video_relevant_recommend(video_id, mid, uid, size, app_type, ab_exp_info):
         top_K = config_.K
         flow_pool_P = config_.P
     else:
-        exp_item_id_list = [int(item.get('expItemId')) for item in ab_exp_info if item is not None]
+        exp_item_id_list = [item.get('expItemId') for item in ab_exp_info]
         if config_.EXP_ITEM_ID['rec_size'] in exp_item_id_list:
             config_value = ab_exp_info[exp_item_id_list.index(config_.EXP_ITEM_ID['rec_size'])].get('configValue')
             log_.info(f'config_value: {config_value}, type: {type(config_value)}')