Skip to content

Commit 2d4e688

Browse files
committed
fix order
1 parent 439aae4 commit 2d4e688

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

onnx_diagnostic/helpers/log_helper.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -594,13 +594,14 @@ def view(
594594

595595
# reorder
596596
if view_def.order:
597-
assert set(view_def.order) <= set_key_columns, (
597+
corder = self._filter_column(view_def.order, all_cols)
598+
assert set(corder) <= set_key_columns, (
598599
f"view_def.name={view_def.name!r}, "
599600
f"non existing columns from order in key_columns "
600-
f"{set(view_def.order) - set_key_columns}"
601+
f"{set(corder) - set_key_columns}"
601602
)
602603
key_columns = [
603-
*[o for o in view_def.order if o in key_columns],
604+
*[o for o in corder if o in key_columns],
604605
*[c for c in key_columns if c not in view_def.order],
605606
]
606607
if view_def.dropna:

0 commit comments

Comments
 (0)