Skip to content

Commit 0955d1b

Browse files
committed
fix aggregation
1 parent bd02ff5 commit 0955d1b

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

_unittests/ut_torch_export_patches/test_patch_transformers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,7 @@ def test_qwen2_5_vl_vision_attention_iteration(self):
428428
f"test_qwen2_5_vl_vision_attention_iteration.{exporter}.onnx"
429429
)
430430
to_onnx(model, inputs, dynamic_shapes=ds, exporter=exporter, filename=filename)
431+
# exporter_kwargs={"report":True} if exporter != "custom" else {}
431432
self.assert_onnx_disc(
432433
f"test_qwen2_5_vl_vision_attention_iteration-{exporter}",
433434
onnx.load(filename),

onnx_diagnostic/helpers/log_helper.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1053,11 +1053,11 @@ def _fix_aggregation_change(
10531053
if select_agg.shape[0] == 0:
10541054
# nothing to fix
10551055
return data
1056-
assert select_agg[columns_to_fix].max() <= 1, (
1057-
f"Column {columns_to_fix!r} has two distinct values at least for one date, "
1058-
f"max={select_agg[columns_to_fix].max()}\n"
1059-
f"{select_agg[select_agg[columns_to_fix] > 1]}"
1060-
)
1056+
# assert select_agg[columns_to_fix].max() <= 1, (
1057+
# f"Column {columns_to_fix!r} has two distinct values at least for one date, "
1058+
# f"max={select_agg[columns_to_fix].max()}\n"
1059+
# f"{select_agg[select_agg[columns_to_fix] > 1]}"
1060+
# )
10611061
return res
10621062

10631063
def _dropna(

0 commit comments

Comments
 (0)