Browse Source

修改变量名

xueyiming 1 day ago
parent
commit
36be204fc9
1 changed files with 4 additions and 4 deletions
  1. 4 4
      ad/pai_flow_operator.py

+ 4 - 4
ad/pai_flow_operator.py

@@ -388,7 +388,7 @@ def extract_date_yyyymmdd(input_string):
         return matches[0]
     return None
 
-def extract_model_version(input_string):
+def extract_model_name(input_string):
     pattern = r"ad_rank_dnn_([^/]+)/\d{8}"
     matches = re.findall(pattern, input_string)
     if matches:
@@ -405,8 +405,8 @@ def get_online_model_config(service_name: str):
     model_config['model_path'] = model_path
     online_date = extract_date_yyyymmdd(model_path)
     model_config['online_date'] = online_date
-    model_version = extract_model_version(model_path)
-    model_config['model_version'] = model_version
+    model_name = extract_model_name(model_path)
+    model_config['model_name'] = model_name
     return model_config
 
 
@@ -799,7 +799,7 @@ if __name__ == '__main__':
             update_online_res = update_online_model()
             if update_online_res:
                 online_model_config = get_online_model_config('ad_rank_dnn_v11_easyrec')
-                update_trained_cids_pointer(online_model_config['model_version'], online_model_config['online_date'])
+                update_trained_cids_pointer(online_model_config['model_name'], online_model_config['online_date'])
                 print("success")
         step_end_time = int(time.time())
         elapsed = step_end_time - start_time