Skip to content

Commit 259de4b

Browse files
committed
Arm backend: Update for lintrunner issue.
1 parent 9ac84b2 commit 259de4b

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/arm/operators/op_maximum.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
from executorch.backends.arm.tosa_specification import TosaSpecification
2323
from executorch.backends.arm.tosa_utils import tosa_shape
2424
from torch.fx import Node
25-
from tosa.NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
2625

2726

2827
@register_node_visitor
@@ -112,6 +111,7 @@ def define_node(
112111
) -> None:
113112

114113
import serializer.tosa_serializer as ts # type: ignore
114+
from tosa import NanPropagationMode # type: ignore[import-untyped]
115115

116116
if inputs[0].dtype != inputs[1].dtype and inputs[0].dtype != output.dtype:
117117
raise TypeError(

backends/arm/operators/op_minimum.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
from executorch.backends.arm.tosa_specification import TosaSpecification
2121
from executorch.backends.arm.tosa_utils import tosa_shape
2222
from torch.fx import Node
23-
from tosa.NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
2423

2524

2625
@register_node_visitor
@@ -110,6 +109,7 @@ def define_node(
110109
) -> None:
111110

112111
import serializer.tosa_serializer as ts # type: ignore
112+
from tosa import NanPropagationMode # type: ignore[import-untyped]
113113

114114
if inputs[0].dtype != inputs[1].dtype and inputs[0].dtype != output.dtype:
115115
raise TypeError(

0 commit comments

Comments
 (0)