Adjust numpy arrays to actual length

JuDFTteam · Jan 24, 2022 · 3ccf158 · 3ccf158
1 parent 106fb6e
commit 3ccf158
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/masci_tools/io/parsers/tabulator/tabulator.py b/masci_tools/io/parsers/tabulator/tabulator.py
@@ -237,6 +237,7 @@ def tabulate(self,
 
         keypaths: KeyPaths = []
 
+        dtypes_set: frozenset[str] = frozenset()
         for index, item in enumerate(collection):
 
             # get inc/ex lists. assume that they are in valid keypaths format already
@@ -269,6 +270,11 @@ def tabulate(self,
                               dtypes=dtypes_set,
                               pass_item_to_transformer=pass_item_to_transformer,
                               **kwargs)
+            length = index + 1
+
+        #Adjust to actual length
+        for column in dtypes_set:
+            table[column] = table[column][:length]
 
         if drop_empty_columns:
             empty_columns = [colname for colname, values in table.items() if all(v is None for v in values)]