Skip to content

Commit d3a2064

Browse files
sneaxiyjinle.zjl
andauthored
Add visual url for math programming task logging (#3110)
* add visual url * add job failure exception Co-authored-by: jinle.zjl <[email protected]>
1 parent c9b6fbf commit d3a2064

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

python/runtime/optimize/optflow.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,10 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
182182
if not query_job_log_url:
183183
raise ValueError("SQLFLOW_OPTFLOW_QUERY_JOB_LOG_URL must be set")
184184

185+
visual_job_url = os.getenv("SQLFLOW_OPTFLOW_VISUAL_JOB_URL")
186+
if not visual_job_url:
187+
raise ValueError("SQLFLOW_OPTFLOW_VISUAL_JOB_URL must be set")
188+
185189
bucket_name = "sqlflow-optflow-models"
186190
bucket = get_bucket(bucket_name)
187191
try:
@@ -223,8 +227,10 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
223227
if not response_json['success']:
224228
raise ValueError("Job submission fails")
225229

226-
print('Job submission succeeds')
227230
record_id = response_json['data']['recordId']
231+
print('Job submission succeeds, record id {}'.format(record_id))
232+
print('FSL URL: {}'.format(fsl_url))
233+
print('Please see log on: {}/{}'.format(visual_job_url, record_id))
228234
try:
229235
success = print_job_log_till_finish(query_job_status_url,
230236
query_job_log_url, record_id,
@@ -233,7 +239,7 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
233239
print("Job succeeds. Save solved result in {}.".format(
234240
result_table))
235241
else:
236-
print("Job fails.")
242+
raise ValueError("Job fails.")
237243
except: # noqa: E722
238244
# FIXME(sneaxiy): we should not delete object if there is any
239245
# network error when querying job status and logs. But when

0 commit comments

Comments
 (0)