Skip to content

Commit e5c1a5d

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 ec1f6a5 commit e5c1a5d

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
@@ -37,7 +37,7 @@
3737
from torch.fx.node import Node
3838

3939
from torch.overrides import TorchFunctionMode
40-
from tosa.TosaGraph import TosaGraph # type: ignore[import-untyped]
40+
from tosa.TosaGraph import TosaGraph # type: ignore[import-not-found, import-untyped]
4141

4242
logger = logging.getLogger(__name__)
4343

@@ -767,7 +767,7 @@ def run_tosa_graph(
767767
inputs_np = [torch_tensor_to_numpy(input_tensor) for input_tensor in inputs]
768768

769769
if isinstance(tosa_version, Tosa_1_00):
770-
import tosa_reference_model as reference_model # type: ignore[import-untyped]
770+
import tosa_reference_model as reference_model # type: ignore[import-not-found, import-untyped]
771771

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

0 commit comments

Comments
 (0)