Add slab to legacy mode

MLSTRUCT · Aug 19, 2024 · 32c54fe · 32c54fe
1 parent 7b8c9dc
commit 32c54fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MLStructFP/db/_db_loader.py b/MLStructFP/db/_db_loader.py
@@ -190,7 +190,7 @@ def tabulate(self, limit: int = 0, legacy: bool = False,
                 ('Project ID', 'Project label', 'Floor ID', 'Cat', 'Elev',
                  'Rects', 'Points', 'Slabs', 'Rooms', 'Items', 'Floor image path'
                  ) if not legacy else
-                ('Floor ID', 'Rects', 'Floor image path')
+                ('Floor ID', 'Rects', 'Slabs', 'Floor image path')
         ):
             theads.append(t)
         table = [theads]
@@ -205,7 +205,7 @@ def tabulate(self, limit: int = 0, legacy: bool = False,
                     (f.project_id, f.project_label, f.id, f.category, 1 if f.elevation else 0,
                      len(f.rect), len(f.point), len(f.slab), len(f.room), len(f.item), f_file
                      ) if not legacy else
-                    (f.id, len(f.rect), f_file)
+                    (f.id, len(f.rect), len(f.slab), f_file)
             ):
                 table_data.append(i)
             table.append(table_data)