Skip to content

Commit 8aa7965

Browse files
Potential fix for code scanning alert no. 32379: Module is imported with 'import' and 'import from'
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
1 parent 500c842 commit 8aa7965

File tree

1 file changed

+10
-11
lines changed

1 file changed

+10
-11
lines changed

onnxruntime/test/testdata/test_evil_weights.py

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import onnx
2-
from onnx import TensorProto, helper
32

43

54
def create_exp_model():
@@ -8,18 +7,18 @@ def create_exp_model():
87
tensors = []
98
outputs = []
109

11-
input_ = helper.make_tensor_value_info("input", TensorProto.INT64, [None, None])
10+
input_ = onnx.helper.make_tensor_value_info("input", onnx.TensorProto.INT64, [None, None])
1211
inputs.append(input_)
1312

14-
evil_tensor = helper.make_tensor(
13+
evil_tensor = onnx.helper.make_tensor(
1514
name="evil_weights",
16-
data_type=TensorProto.INT64,
15+
data_type=onnx.TensorProto.INT64,
1716
# dims=[100, 100],
1817
dims=[10],
1918
vals=[],
2019
)
2120
tensors.append(evil_tensor)
22-
evil_tensor.data_location = TensorProto.EXTERNAL
21+
evil_tensor.data_location = onnx.TensorProto.EXTERNAL
2322
entry1 = evil_tensor.external_data.add()
2423
entry1.key = "location"
2524
entry1.value = "*/_ORT_MEM_ADDR_/*"
@@ -31,15 +30,15 @@ def create_exp_model():
3130
entry3.key = "length"
3231
entry3.value = "80"
3332

34-
tensors.append(helper.make_tensor(name="0x1", data_type=TensorProto.INT64, dims=[1], vals=[0x1]))
35-
nodes.append(helper.make_node(op_type="Add", inputs=["evil_weights", "0x1"], outputs=["output"]))
33+
tensors.append(onnx.helper.make_tensor(name="0x1", data_type=onnx.TensorProto.INT64, dims=[1], vals=[0x1]))
34+
nodes.append(onnx.helper.make_node(op_type="Add", inputs=["evil_weights", "0x1"], outputs=["output"]))
3635

37-
outputs.append(helper.make_tensor_value_info("output", TensorProto.INT64, []))
36+
outputs.append(onnx.helper.make_tensor_value_info("output", onnx.TensorProto.INT64, []))
3837

39-
graph = helper.make_graph(nodes, "test", inputs, outputs, tensors)
40-
model = helper.make_model(
38+
graph = onnx.helper.make_graph(nodes, "test", inputs, outputs, tensors)
39+
model = onnx.helper.make_model(
4140
graph,
42-
opset_imports=[helper.make_opsetid("", 18), helper.make_opsetid("ai.onnx.ml", 3)],
41+
opset_imports=[onnx.helper.make_opsetid("", 18), onnx.helper.make_opsetid("ai.onnx.ml", 3)],
4342
ir_version=11,
4443
)
4544

0 commit comments

Comments
 (0)