Skip to content

Commit 5df0ddc

Browse files
committed
fix
1 parent 2d4e688 commit 5df0ddc

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

onnx_diagnostic/helpers/log_helper.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -594,7 +594,8 @@ def view(
594594

595595
# reorder
596596
if view_def.order:
597-
corder = self._filter_column(view_def.order, all_cols)
597+
subset = self._filter_column(view_def.order, all_cols | {self.time})
598+
corder = [o for o in view_def.order if o in subset]
598599
assert set(corder) <= set_key_columns, (
599600
f"view_def.name={view_def.name!r}, "
600601
f"non existing columns from order in key_columns "
@@ -835,7 +836,7 @@ def to_excel(
835836
print(f"[CubeLogs.to_excel] saving sheet {name!r} in {name_csv!r}")
836837
df.reset_index(drop=False).to_csv(f"{output}.{name}.csv", index=False)
837838

838-
if memory > 2**21:
839+
if memory > 2**22:
839840
msg = (
840841
f"[CubeLogs.to_excel] skipping {name!r}, "
841842
f"too big for excel {memory} bytes"

0 commit comments

Comments
 (0)