DL: Replace e.args[0] with e.message
JIRA: MADLIB-1467
Previously we were relying on e.args[0] in our except block but some of the
exceptions raised by tensorflow like `InvalidArgumentError` do not have e.args
as empty. This is not usual but there is no guarantee of args being populated
so we should instead rely on e.message(typically e.args[0] is the same as
e.message)
Co-authored-by: Ekta Khanna<ekhanna@vmware.com>
diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras.sql_in b/src/ports/postgres/modules/deep_learning/madlib_keras.sql_in
index da61e8e..2620706 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras.sql_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras.sql_in
@@ -1691,7 +1691,7 @@
except Exception as e:
etype, _, tb = exc_info()
detail = ''.join(traceback.format_exception(etype, e, tb))
- message = e.args[0] + 'TransAggDetail' + detail
+ message = e.message + 'TransAggDetail' + detail
e.args = (message,)
raise e
$$ LANGUAGE plpythonu
@@ -1733,7 +1733,7 @@
except Exception as e:
etype, _, tb = exc_info()
detail = ''.join(traceback.format_exception(etype, e, tb))
- message = e.args[0] + 'TransAggDetail' + detail
+ message = e.message + 'TransAggDetail' + detail
e.args = (message,)
raise e
$$ LANGUAGE plpythonu
@@ -1753,7 +1753,7 @@
except Exception as e:
etype, _, tb = exc_info()
detail = ''.join(traceback.format_exception(etype, e, tb))
- message = e.args[0] + 'MergeAggDetail' + detail
+ message = e.message + 'MergeAggDetail' + detail
e.args = (message,)
raise e
$$ LANGUAGE plpythonu
@@ -1771,7 +1771,7 @@
except Exception as e:
etype, _, tb = exc_info()
detail = ''.join(traceback.format_exception(etype, e, tb))
- message = e.args[0] + 'FinalAggDetail' + detail
+ message = e.message + 'FinalAggDetail' + detail
e.args = (message,)
raise e
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 42fd7d9..84e4426 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.args[0] + '\nTransAggDetail:\n' + detail
+ message = e.message + '\nTransAggDetail:\n' + detail
e.args = (message,)
raise e
$$ LANGUAGE plpythonu
