[madlib] 02/02: DL: Fix object table schema error message

[kh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

khannaekta pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/madlib.git

commit 77f3fd7ef6fdc0eb54656d73ae0d2c06244062ad
Author: Orhan Kislal <ok...@apache.org>
AuthorDate: Fri Apr 30 18:30:00 2021 +0300

    DL: Fix object table schema error message
---
 .../modules/deep_learning/madlib_keras_validator.py_in         | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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 @@ class FitCommonValidator(object):
             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.")