liqian 2 년 전
부모
커밋
bef0ca4873
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      ad_user_video_predict.py

+ 4 - 1
ad_user_video_predict.py

@@ -1,5 +1,5 @@
 import datetime
-
+import sys
 import numpy as np
 import pandas as pd
 from odps import ODPS
@@ -84,6 +84,9 @@ def predict_ad_group_video(dt, config_key, config_param):
     video_df = video_df[video_df['videoid'] != -1]
     log_.info(f"video_df count = {len(video_df)}")
 
+    if len(group_df) == 0 or len(video_df):
+        sys.exit(1)
+
     predict_df = video_df
     all_group_data = []
     for index, item in group_df.iterrows():