Skip to content

Commit b771e94

Browse files
committed
fixes
1 parent d1cf66f commit b771e94

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

onnx_diagnostic/helpers/log_helper.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1921,9 +1921,7 @@ def first_err(df: pandas.DataFrame) -> pandas.Series:
19211921
return lambdas[formula]
19221922

19231923
if formula == "onnx_n_nodes_no_cst":
1924-
return lambda df: gdf(df, "onnx_n_nodes", 0) - gdf(
1925-
df, "op_onnx__Constant", 0
1926-
).fillna(0)
1924+
return lambda df: gdf(df, "onnx_n_nodes", 0) - gdf(df, "op_onnx__Constant", 0)
19271925
if formula == "peak_gpu_torch":
19281926
return lambda df: gdf(df, "mema_gpu_5_after_export") - gdf(df, "mema_gpu_4_reset")
19291927
if formula == "peak_gpu_nvidia":

0 commit comments

Comments
 (0)