remove hardcode of online inference id test

paul666chen · Jan 6, 2022 · 0f9b418 · 0f9b418
1 parent 8fdfc27
commit 0f9b418
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/examples/flow_test_template/hetero_lr/flow_test_config.yaml b/examples/flow_test_template/hetero_lr/flow_test_config.yaml
@@ -3,6 +3,7 @@ api_version: v1
 component_name: hetero_lr_0
 metric_output_path: ./output/metric/{}_hetero_lr_0.json
 model_output_path: ./output/model/{}_hetero_lr_0.json
+phone_num: 12345678900
 
 flow_test_template:
   train_conf_path: /examples/flow_test_template/hetero_lr/hetero_lr_normal_conf.json

diff --git a/examples/flow_test_template/homo_lr/flow_test_config.yaml b/examples/flow_test_template/homo_lr/flow_test_config.yaml
@@ -4,6 +4,7 @@ component_name: homo_lr_0
 component_is_homo: true
 metric_output_path: ./output/metric/{}_homo_lr_0.json
 model_output_path: ./output/model/{}_homo_lr_0.json
+phone_num: 12345678900
 
 flow_test_template:
   train_conf_path: /examples/flow_test_template/homo_lr/homo_lr_train_conf.json

diff --git a/python/fate_test/fate_test/flow_test/flow_process.py b/python/fate_test/fate_test/flow_test/flow_process.py
@@ -320,19 +320,19 @@ def bind_model(self):
                             f"details: {resp_data.get('retmsg')}")
         raise Exception(f"Request model bind api failed, status code: {response.status_code}")
 
-    def online_predict(self, online_serving):
+    def online_predict(self, online_serving, phone_num):
         serving_url = f"http://{online_serving}/federation/1.0/inference"
         post_data = {
             "head": {
                 "serviceId": self.service_id
             },
             "body": {
                 "featureData": {
-                    "phone_num": "18576635456",
+                    "phone_num": phone_num,
                 },
                 "sendToRemoteFeatureData": {
                     "device_type": "imei",
-                    "phone_num": "18576635456",
+                    "phone_num": phone_num,
                     "encrypt_type": "raw"
                 }
             }
@@ -360,6 +360,7 @@ def run_fate_flow_test(config_json):
     metric_output_path = config_json['metric_output_path']
     model_output_path = config_json['model_output_path']
     serving_connect_bool = serving_connect(config_json['serving_setting'])
+    phone_num = config_json['phone_num']
 
     print('submit train job')
     # train
@@ -400,4 +401,4 @@ def run_fate_flow_test(config_json):
         utilize.bind_model()
 
         # online predict
-        utilize.online_predict(online_serving=online_serving)
+        utilize.online_predict(online_serving=online_serving, phone_num=phone_num)
diff --git a/python/fate_test/fate_test/scripts/flow_test_cli.py b/python/fate_test/fate_test/scripts/flow_test_cli.py
@@ -162,6 +162,7 @@ def get_role(conf: Config):
                    'model_file_path': os.path.abspath(conf.data_base_dir) + flow_test_template['model_conf_path'],
                    'server_url': "http://{}/{}".format(flow_services, config['api_version']),
                    'train_auc': config['train_auc'],
+                   'phone_num': config['phone_num'],
                    'component_name': config['component_name'],
                    'component_is_homo': config.get('component_is_homo', False),
                    'serving_setting': conf.serving_setting['serving_setting']['address'],