Skip to content

Commit 360321e

Browse files
committed
Identify all columns that start with 'exception_info' so that they can be dropped
1 parent 5879c6a commit 360321e

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

scripts/jobs/housing/housing_apply_gx_dq_tests.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)