1. rename .env to fate.env

2. If the save pipelined model fails, the job fails Signed-off-by: zengjice <[email protected]>
pyisong · May 20, 2020 · 0bd9204 · 0bd9204
1 parent ecf08dc
commit 0bd9204
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/.env → fate.env b/.env → fate.env
diff --git a/fate_flow/driver/task_scheduler.py b/fate_flow/driver/task_scheduler.py
@@ -469,6 +469,8 @@ def finish_job(job_id, job_runtime_conf, stop=False):
                                              src_role=job_initiator['role'],
                                              json_body={},
                                              work_mode=job_parameters['work_mode'])
+                    if response['retcode'] != 0:
+                        raise Exception('job {} save pipeline failed on role {} party {}'.format(job_id, role, party_id))
                 # clean
                 federated_api(job_id=job_id,
                               method='POST',

diff --git a/fate_flow/entity/runtime_config.py b/fate_flow/entity/runtime_config.py
@@ -42,7 +42,7 @@ def init_config(**kwargs):
 
     @staticmethod
     def init_env():
-        RuntimeConfig.ENV.update(dotenv.dotenv_values(dotenv_path=os.path.join(get_project_base_directory(), ".env")))
+        RuntimeConfig.ENV.update(dotenv.dotenv_values(dotenv_path=os.path.join(get_project_base_directory(), "fate.env")))
 
     @staticmethod
     def get_env(key):