Skip to content

Commit 3f02fec

Browse files
committed
nit: fix tmp_path and updated model name
Signed-off-by: gcunhase <[email protected]>
1 parent 55cea1a commit 3f02fec

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/_test_utils/onnx_quantization/lib_test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def build_matmul_relu_model_ir_12():
417417
]
418418

419419
# Create the ONNX graph with the nodes and initializers
420-
graph = helper.make_graph(nodes, "r1a", inputs, outputs, initializer=initializers)
420+
graph = helper.make_graph(nodes, "matmul_relu", inputs, outputs, initializer=initializers)
421421

422422
# Create the ONNX model
423423
model = helper.make_model(graph)

tests/unit/onnx/test_onnx_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def test_remove_node_extra_training_outputs():
257257
assert "saved_inv_std" not in value_info_names
258258

259259

260-
def test_ir_version_support(tmp_path="./"):
260+
def test_ir_version_support(tmp_path):
261261
model = build_matmul_relu_model_ir_12()
262262
model_path = os.path.join(tmp_path, "test_matmul_relu.onnx")
263263
onnx.save(model, model_path)

0 commit comments

Comments
 (0)