DL: Fix object table schema error message
diff --git a/src/ports/postgres/modules/deep_learning/madlib_keras_validator.py_in b/src/ports/postgres/modules/deep_learning/madlib_keras_validator.py_in
index de5c63d..b809477 100644
--- a/src/ports/postgres/modules/deep_learning/madlib_keras_validator.py_in
+++ b/src/ports/postgres/modules/deep_learning/madlib_keras_validator.py_in
@@ -327,7 +327,15 @@
input_tbl_valid(self.validation_summary_table, self.module_name)
if self.object_table is not None:
- input_tbl_valid(self.object_table, self.module_name)
+ try:
+ input_tbl_valid(self.object_table, self.module_name)
+ except Exception as e:
+ if "Incorrect table name" in e:
+ plpy.error("""Incorrect table name ({0}) provided! """\
+ """Do not include schema name for the object table.
+ """.format(self.object_table))
+ else:
+ raise e
_assert(is_superuser(get_table_owner(self.object_table)),
"DL: Cannot use a table of a non-superuser as object table.")