File tree 1 file changed +3
-1
lines changed 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,7 @@ class TableEvaluation(UnitEvaluation):
32
32
table_id : int = - 1
33
33
TEDS : float
34
34
is_complex : bool = False
35
+ is_structure_only : bool = False
35
36
36
37
true_ncols : int = - 1
37
38
pred_ncols : int = - 1
@@ -227,7 +228,7 @@ def __call__(
227
228
dataset_evaluation = DatasetTableEvaluation (
228
229
evaluated_samples = len (table_evaluations ),
229
230
rejected_samples = rejected_samples ,
230
- evaluations = table_evaluations ,
231
+ evaluations = table_evaluations + table_struct_evaluations ,
231
232
TEDS = compute_stats (teds_all ),
232
233
TEDS_struct = compute_stats (teds_struct ),
233
234
TEDS_simple = compute_stats (teds_simple ),
@@ -297,6 +298,7 @@ def _evaluate_tables_in_documents(
297
298
pred_ncols = pred_table .data .num_cols ,
298
299
true_nrows = true_table .data .num_rows ,
299
300
pred_nrows = pred_table .data .num_rows ,
301
+ is_structure_only = structure_only ,
300
302
)
301
303
table_evaluations .append (table_evaluation )
302
304
except Exception :
You can’t perform that action at this time.
0 commit comments