[Bug] stop sending report when no-such-table (#894)

* stop sending report when no-such-table Signed-off-by: Ching Yi, Chan <[email protected]> * Only ignore table-not-found Signed-off-by: Ching Yi, Chan <[email protected]> --------- Signed-off-by: Ching Yi, Chan <[email protected]>
InfuseAI · Oct 5, 2023 · 6e36aa6 · 6e36aa6
1 parent 893b28a
commit 6e36aa6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/piperider_cli/profiler/profiler.py b/piperider_cli/profiler/profiler.py
@@ -17,6 +17,7 @@
 from sqlalchemy.sql.elements import ColumnClause
 from sqlalchemy.sql.expression import CTE, false, true, table as table_clause, column as column_clause
 from sqlalchemy.types import Float
+from sqlalchemy.exc import NoSuchTableError
 
 from .event import ProfilerEventHandler, DefaultProfilerEventHandler
 from ..configuration import Configuration
@@ -163,10 +164,11 @@ def _fetch_table_task(subject: ProfileSubject):
                 table = None
                 try:
                     table = Table(subject.table, MetaData(), autoload_with=engine, schema=schema)
-                except Exception as e:
+                except NoSuchTableError:
                     # ignore the table metadata fetch error
-                    capture_exception(e)
                     pass
+                except Exception as e:
+                    capture_exception(e)
                 return subject, table
 
             future = _run_in_executor(self.executor, _fetch_table_task, subject)