Skip to content

Commit 858a2d5

Browse files
committed
Arm backend: Use correct ignore for import-not-found
Add ignore for import as well, since on github these packages are not installed and thus its a import-not-found error and not untyped as it becomes locally if you have the packages installed. Signed-off-by: per.held@arm.com Change-Id: Iba7b579f8f95991aca26054698e1d357dc14a853
1 parent 7b209a6 commit 858a2d5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/arm/test/runner_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
from torch.fx.node import Node
3737

3838
from torch.overrides import TorchFunctionMode
39-
from tosa.TosaGraph import TosaGraph # type: ignore[import-untyped]
39+
from tosa.TosaGraph import TosaGraph # type: ignore[import-not-found, import-untyped]
4040

4141
logger = logging.getLogger(__name__)
4242

@@ -762,7 +762,7 @@ def run_tosa_graph(
762762
inputs_np = [torch_tensor_to_numpy(input_tensor) for input_tensor in inputs]
763763

764764
if isinstance(tosa_version, Tosa_1_00):
765-
import tosa_reference_model as reference_model # type: ignore[import-untyped]
765+
import tosa_reference_model as reference_model # type: ignore[import-not-found, import-untyped]
766766

767767
debug_mode = "ALL" if logger.getEffectiveLevel() <= logging.DEBUG else None
768768
outputs_np, status = reference_model.run(

0 commit comments

Comments
 (0)