Skip to content

Commit f114eb9

Browse files
gramalingamCopilot
andcommitted
Fix mixed import style for onnxscript in test files
Use "from onnxscript import values" instead of "import onnxscript" to avoid mixing import and from-import for the same module. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent 6a6e6f7 commit f114eb9

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

onnxscript/rewriter/ort_fusions/mha_bias_test.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,13 @@
1414
import numpy as np
1515
import onnx_ir as ir
1616

17-
import onnxscript
18-
import onnxscript.rewriter.ort_fusions._test_utils as test_utils
19-
from onnxscript import FLOAT, INT32, script
17+
from onnxscript import FLOAT, INT32, script, values
2018
from onnxscript import opset18 as op
2119
from onnxscript.optimizer import optimize
20+
from onnxscript.rewriter.ort_fusions import _test_utils as test_utils
2221
from onnxscript.rewriter.ort_fusions.mha_bias import fuse_mha_bias
2322

24-
msft_op = onnxscript.values.Opset("com.microsoft", 1)
23+
msft_op = values.Opset("com.microsoft", 1)
2524

2625
_B, _S, _D, _Dk, _Dv = 2, 8, 16, 16, 16
2726
_NUM_HEADS = 4

onnxscript/rewriter/ort_fusions/mha_scale_test.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,13 @@
1515
import numpy as np
1616
import onnx_ir as ir
1717

18-
import onnxscript
19-
import onnxscript.rewriter.ort_fusions._test_utils as test_utils
20-
from onnxscript import FLOAT, script
18+
from onnxscript import FLOAT, script, values
2119
from onnxscript import opset18 as op
2220
from onnxscript.optimizer import optimize
21+
from onnxscript.rewriter.ort_fusions import _test_utils as test_utils
2322
from onnxscript.rewriter.ort_fusions.mha_scale import fuse_mha_scale
2423

25-
msft_op = onnxscript.values.Opset("com.microsoft", 1)
24+
msft_op = values.Opset("com.microsoft", 1)
2625

2726
_B, _S, _D = 2, 8, 16
2827
_NUM_HEADS = 4

0 commit comments

Comments
 (0)