ソースを参照

Merge branch 'update-sql-test' into pre-master

liqian 3 年 前
コミット
59c109b1b5
2 ファイル変更5 行追加3 行削除
  1. 1 1
      rov_train.py
  2. 4 2
      utils.py

+ 1 - 1
rov_train.py

@@ -220,6 +220,6 @@ if __name__ == '__main__':
 
     log_.info('rov model predict start...')
     predict_start = time.time()
-    predict()
+    predict_test()
     predict_end = time.time()
     log_.info('rov model predict end, execute time = {}ms'.format((predict_end - predict_start)*1000))

+ 4 - 2
utils.py

@@ -99,7 +99,8 @@ def filter_video_status(video_ids):
     :return: filtered_videos
         """
     if len(video_ids) == 1:
-        sql = "SELECT video_id " \
+        sql = "set hg_experimental_enable_shard_pruning=off; " \
+              "SELECT video_id " \
               "FROM {} " \
               "WHERE audit_status = 5 " \
               "AND applet_rec_status IN (1, -6) " \
@@ -109,7 +110,8 @@ def filter_video_status(video_ids):
               "AND transcoding_status = 3 " \
               "AND video_id IN ({});".format(config_.VIDEO_STATUS, video_ids[0])
     else:
-        sql = "SELECT video_id " \
+        sql = "set hg_experimental_enable_shard_pruning=off; " \
+              "SELECT video_id " \
               "FROM {} " \
               "WHERE audit_status = 5 " \
               "AND applet_rec_status IN (1, -6) " \