Skip to content

Commit a0d7d08

Browse files
committed
nit: fix tmp_path and updated model name
Signed-off-by: gcunhase <[email protected]>
1 parent 7b764ff commit a0d7d08

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
@@ -600,7 +600,7 @@ def build_matmul_relu_model_ir_12():
600600
]
601601

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

605605
# Create the ONNX model
606606
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)