Catch evaluate model data error

gretelai · Sep 6, 2023 · 8dd2770 · 8dd2770
1 parent 5b03a30
commit 8dd2770
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/src/gretel_trainer/relational/multi_table.py b/src/gretel_trainer/relational/multi_table.py
@@ -1033,11 +1033,19 @@ def generate(
             if table not in self.relational_data.list_all_tables(Scope.EVALUATABLE):
                 continue
 
-            evaluate_data = self._strategy.get_evaluate_model_data(
-                rel_data=self.relational_data,
-                table_name=table,
-                synthetic_tables=output_tables,
-            )
+            try:
+                evaluate_data = self._strategy.get_evaluate_model_data(
+                    rel_data=self.relational_data,
+                    table_name=table,
+                    synthetic_tables=output_tables,
+                )
+            except:
+                logger.error(
+                    "Error getting evaluate model data for table %s",
+                    table,
+                    exc_info=True,
+                )
+                evaluate_data = None
             if evaluate_data is None:
                 continue