DL: Fix exception handling bug in fit multiple
JIRA: MADLIB-1467
When catching an exception in fit multiple, we were splitting by the wrong
keyword. This commit fixes the bug and also removes '_' from the keyword name.
Co-authored-by: Ekta Khanna<ekhanna@vmware.com>
diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
index aa7a2bc..2db346e 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.py_in
@@ -953,9 +953,9 @@
plpy_execute(self.udf_plan, [ self.is_final_training_call ] )
except plpy.SPIError as e:
msg = e.message
- if not 'UDF_Detail' in msg:
+ if not 'UDFDetail' in msg:
raise e
- e.message, detail = msg.split('UDF_Detail')
+ e.message, detail = msg.split('UDFDetail')
# Extract Traceback from segment, add to
# DETAIL of error message on coordinator
e.args = (e.message,)
diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.sql_in b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.sql_in
index 84e4426..6d10da4 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.sql_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras_fit_multiple_model.sql_in
@@ -1539,7 +1539,7 @@
except Exception as e:
etype, _, tb = exc_info()
detail = ''.join(traceback.format_exception(etype, e, tb))
- message = e.message + '\nTransAggDetail:\n' + detail
+ message = e.message + 'UDFDetail' + detail
e.args = (message,)
raise e
$$ LANGUAGE plpythonu
