@@ -182,6 +182,10 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
182
182
if not query_job_log_url :
183
183
raise ValueError ("SQLFLOW_OPTFLOW_QUERY_JOB_LOG_URL must be set" )
184
184
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
+
185
189
bucket_name = "sqlflow-optflow-models"
186
190
bucket = get_bucket (bucket_name )
187
191
try :
@@ -223,8 +227,10 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
223
227
if not response_json ['success' ]:
224
228
raise ValueError ("Job submission fails" )
225
229
226
- print ('Job submission succeeds' )
227
230
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 ))
228
234
try :
229
235
success = print_job_log_till_finish (query_job_status_url ,
230
236
query_job_log_url , record_id ,
@@ -233,7 +239,7 @@ def submit_optflow_job(train_table, result_table, fsl_file_content, solver,
233
239
print ("Job succeeds. Save solved result in {}." .format (
234
240
result_table ))
235
241
else :
236
- print ("Job fails." )
242
+ raise ValueError ("Job fails." )
237
243
except : # noqa: E722
238
244
# FIXME(sneaxiy): we should not delete object if there is any
239
245
# network error when querying job status and logs. But when
0 commit comments