Skip to content

Commit 9e47e33

Browse files
committed
fix command lien
1 parent 1eeedc4 commit 9e47e33

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

onnx_diagnostic/_command_lines_parser.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1180,7 +1180,7 @@ def _cmd_sbs(argv: List[Any]):
11801180
import pandas
11811181
import torch
11821182
from .helpers import string_type
1183-
from .torch_onnx.sbs import run_aligned, post_process_run_aligned_obs
1183+
from .torch_onnx.sbs import run_aligned
11841184
from .reference import OnnxruntimeEvaluator
11851185

11861186
parser = get_parser_sbs()
@@ -1241,9 +1241,12 @@ def _size(name):
12411241
use_tensor=True,
12421242
exc=False,
12431243
):
1244-
pobs = post_process_run_aligned_obs(obs)
1245-
data.append(pobs)
1246-
if "initializer" not in pobs and "placeholder" not in pobs and len(data) % ratio == 0:
1244+
data.append(obs)
1245+
if (
1246+
obs.onnx_op_type != "initializer"
1247+
and onnx.ep_target != "placeholder"
1248+
and len(data) % ratio == 0
1249+
):
12471250
df = pandas.DataFrame(data).apply(
12481251
lambda col: col.fillna("") if col.dtype == "object" else col
12491252
)

0 commit comments

Comments
 (0)