File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -107,10 +107,8 @@ def main():
107107 checkpoint_result = checkpoint .run (batch_parameters = batch_parameters )
108108 results_dict = list (checkpoint_result .run_results .values ())[0 ].to_json_dict ()
109109 table_results_df = pd .json_normalize (results_dict ['results' ])
110- table_results_df = table_results_df .drop (
111- columns = {'exception_info.raised_exception' ,
112- 'exception_info.exception_traceback' ,
113- 'exception_info.exception_message' })
110+ cols_to_drop = [c for c in table_results_df .columns if c .startswith ('exception_info' )]
111+ table_results_df = table_results_df .drop (columns = cols_to_drop )
114112 table_results_df_list .append (table_results_df )
115113
116114 # generate id lists for each unexpected result set
You can’t perform that action at this time.
0 commit comments